网络训练报错RuntimeError:size mismatch, m1: [256 x 1600], m2: [1024 x 256]的解决办法

最近在用fashionMNIST跑一个网络,结果以下图python 想要看看用同一个网络,不一样数据集之间对准确率差异,所以我把数据集换成了cifar10.除了要用到transforms.Grayscale(1)网络 test_dataset = CIFAR10('../data/CIFAR10', train=False, download=True, transform=transforms.
相关文章
相关标签/搜索