Pytorch Calculate Loss - backward(), and autograd will do the rest. Although they refer to the running_loss (epoch loss in ...