wyboooo:

Size([8, 3, 24, 24]) torch.Size([8, 3, 96, 96]) 


Here is the complete error message:
Traceback (most recent call last):
  File "train_GFN_4x.py", line 191, in <module>
    train(trainloader, model, criterion, optimizer, epoch, opt.contrast_w)
  File "train_GFN_4x.py", line 118, in train
    loss1 = criterion(lr_deblur, LR_Deblur)
  File "/usr/local/conda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/usr/local/conda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 445, in forward
    return F.mse_loss(input, target, reduction=self.reduction)
  File "/usr/local/conda/envs/py36/lib/python3.6/site-packages/torch/nn/functional.py", line 2647, in mse_loss
    expanded_input, expanded_target = torch.broadcast_tensors(input, target)
  File "/usr/local/conda/envs/py36/lib/python3.6/site-packages/torch/functional.py", line 65, in broadcast_tensors
    return _VF.broadcast_tensors(tensors)
RuntimeError: The size of tensor a (8) must match the size of tensor b (16) at non-singleton dimension 0