What is the difference between those two tensors?
loss1 = tensor(22081814., device='cuda:0', grad_fn=<ThAddBackward>) 
loss2 = tensor(1272513408., device='cuda:0', grad_fn=<SumBackward0>)

They are the loss values to be used for step.backward().