網絡訓練報錯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.

>>阅读原文<<