While running the python code I am getting the following error. Can anyone tell me where I am going wrong?
The input image is just resized from the original image. It works fine with the original image, but when I resize it produces the error.
Error-
RuntimeError: input and target shapes do not match: input [1 x 3 x 64 x 64], target [1 x 1 x 64 x 64] at /opt/conda/conda-bld/pytorch_1532502421238/work/aten/src/THCUNN/generic/MSECriterion.cu:12