Bootstrap

ValueError: pic should be 2/3 dimensional. Got 4 dimensions

在训练神经网络模型的时候,出现了这个错误:

ValueError: pic should be 2/3 dimensional. Got 4 dimensions

将dataset中的数据和标签的transforms=transforms.ToTensor改写为不进行to tnesor这一步即可解决问题。

    # x_transforms = transforms.ToTensor()
    x_transforms=None

原因是你已经进行过to tensor了。

;