a tensor epoch_loss += loss_val.item() # accumulate loss_val.backward() optimizer.step() The statement loss_val = loss_obj(oupt, Y) is slightly misleading because it's using a quirky Python language ...