I have tried to print the .grad_fn and the requires_grad of the model and the loss tensors. On both occasions the loss tensors does not print anything. Not entirely sure If I am getting this right.
I tried print the first iteration of the loss tensor and it displayed tensor(2.155, device:'cuda:0')