当前位置: 代码迷 >> 综合 >> ValueError: pic should be 2/3 dimensional. Got 4 dimensions
  详细解决方案

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

热度:30   发布时间:2024-01-25 15:24:55.0

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

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了。

  相关解决方案