RuntimeError: size mismatch, m1: [2 x 4608], m2: [2048 x 1024] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:249
Myzhencai opened this issue · comments
I have changed the picture to 1920x1080 and 480x270 for trainning but I got this error:
Could you please help me out ,thanks ! Have a nice day :-).
When the img size is different. You may have to change the discriminator structure to grt desired output shape.
@thangvubk thanks for reply but when I changed the classify to 2x4608 I got this error:-)?