Skip to content Skip to sidebar Skip to footer
Showing posts with the label Torchvision

Size Mismatch, M1: [3584 X 28], M2: [784 X 128] At /pytorch/aten/src/TH/generic/THTensorMath.cpp:940

I have executed the following code and getting the error shown at extreme bottom. I would like to k… Read more Size Mismatch, M1: [3584 X 28], M2: [784 X 128] At /pytorch/aten/src/TH/generic/THTensorMath.cpp:940