ref: inner train loop (intermediate step) 3/n

Author: williamFalcon

pytorch_lightning/trainer/training_loop_temp.py

@@ -137,7 +137,7 @@ def get_optimizers_iterable(self):
     def backward(self, result, optimizer, opt_idx):
         # backward pass
         with self.trainer.profiler.profile('model_backward'):
-            result.closure_loss = self.trainer.accelerator_backend.backward(result.loss, optimizer, opt_idx)
+            result.closure_loss = self.trainer.accelerator_backend.backward(result.closure_loss, optimizer, opt_idx)
 
     def on_after_backward(self, training_step_output, batch_idx, untouched_loss):
         is_result_obj = isinstance(training_step_output, Result)