Update pytorch_lightning/loops/optimization/optimizer_loop.py

Author: awaelchli

pytorch_lightning/loops/optimization/optimizer_loop.py

@@ -460,7 +460,7 @@ def _training_step(self, split_batch: Any, batch_idx: int, opt_idx: int) -> Clos
 
             result = ClosureResult.from_training_step_output(training_step_output, self.trainer.accumulate_grad_batches)
 
-            if self.trainer._terminate_on_nan:  # : 212
+            if self.trainer._terminate_on_nan:
                 check_finite_loss(result.closure_loss)
 
             if self.trainer.move_metrics_to_cpu: