Dimension mismatch error on changing image dimensions

[purijs]

My hr images are 256x256 and lr is 64x64, with batch_size=8 and scale=4

I'm getting this error:
RuntimeError: size mismatch, m1: [8 x 73728], m2: [8192 x 100] at /opt/conda/conda-bld/pytorch_1579022060824/work/aten/src/THC/generic/THCTensorMathBlas.cu:290

Before training, you have to resize your hr & lr images. (hr: 128x128, lr: 32x32)