Size mismatched.
I am a beginner and I try to use simple classifier I found online and use my own custom image.
I have 2 class of RGB image with size of 224x224.
When I run below code I get an mismatched size.
May I know how the number came from?
RuntimeError: size mismatch, m1: [2 x 119072], m2: [1568 x 2] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:290
I know 10 came from batch size but how about 100352?
Below is my code and my error
27ogos1158×865 104 KB