Thank you for your kindness. I tried to change “x = x.view(x.size(0), 163737)”.
But I have a new error,“RuntimeError: size mismatch, m1: [4 x 21904], m2: [4000 x 120] at /Users/administrator/nightlies/pytorch-1.0.0/wheel_build_dirs/conda_3.6/conda/conda-bld/pytorch_1544137972173/work/aten/src/TH/generic/THTensorMath.cpp:940”
I tried to search the answer.but I have no idea.
I predict my black and white images to be the cause.