mat1 dim 1 must match mat2 dim 0

[qustions]

hi @ewrfcas there is NO error if I decrease the image size in train.py
If I change lr_height, lr_width and hr_height, hr_width in data_loader.py to my input size it's giving me the error.
But it's working fine with default sizes. what I am missing here

the error is

ret = torch.addmm(bias, input, weight.t())
RuntimeError: mat1 dim 1 must match mat2 dim 0

[ewrfcas]

Which line does it happens? However, if you use the input size different from the default size, you need to retrain the model.