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

发布了8 篇原创文章 · 获赞 2 · 访问量 417

猜你喜欢

转载自blog.csdn.net/weixin_44262126/article/details/104024171
今日推荐