Hi, have you seen this warning in the stack trace:
Warning (from warnings module):
File “C:\Users\Asus\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\nn\modules\loss.py”, line 431
return F.mse_loss(input, target, reduction=self.reduction)
UserWarning: Using a target size (torch.Size([100, 360, 360])) that is different to the input size (torch.Size([100, 2])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
So I think the problem is that what you give to your MSE loss are two tensors of different sizes, so it will give unexpected results.