train_error!

[xaohuicai]

Using a target size (torch.Size([1, 1, 768, 1024])) that is different to the input size (torch.Size([1, 1, 96, 128])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
Traceback (most recent call last):
File "train.py", line 214, in
main()
File "train.py", line 84, in main
train(train_list, model, criterion, optimizer, epoch)
File "train.py", line 136, in train
loss = criterion(output, target)
File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/loss.py", line 443, in forward
return F.mse_loss(input, target, reduction=self.reduction)
File "/usr/local/lib/python3.5/dist-packages/torch/nn/functional.py", line 2256, in mse_loss
expanded_input, expanded_target = torch.broadcast_tensors(input, target)
File "/usr/local/lib/python3.5/dist-packages/torch/functional.py", line 62, in broadcast_tensors
return torch._C._VariableFunctions.broadcast_tensors(tensors)
RuntimeError: The size of tensor a (128) must match the size of tensor b (1024) at non-singleton dimension 3