styling

Author: Swetha Mandava

pytorch_lightning/trainer/training_loop.py

@@ -557,7 +557,10 @@ def run_training_epoch(self):
 
             # hook
             # TODO: add outputs to batches
-            self.on_train_batch_end(epoch_output, batch_output.training_step_output_for_epoch_end, batch, batch_idx, dataloader_idx)
+            self.on_train_batch_end(
+                epoch_output,
+                batch_output.training_step_output_for_epoch_end,
+                batch, batch_idx, dataloader_idx)
 
             # -----------------------------------------
             # SAVE METRICS TO LOGGERS