remove outdated changes

Lightning-AI · Dec 4, 2020 · 9ce9ea0 · 9ce9ea0
1 parent da623a3
commit 9ce9ea0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py b/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py
@@ -541,7 +541,7 @@ def __run_legacy_training_epoch_end(
 
         return epoch_log_metrics, epoch_progress_bar_metrics, epoch_callback_metrics
 
-    def __auto_reduce_results_on_epoch_end(self, epoch_output, device: Optional[Union[str, torch.device]] = None):
+    def __auto_reduce_results_on_epoch_end(self, epoch_output):
         epoch_log_metrics = {}
         epoch_progress_bar_metrics = {}
         for opt_outputs in epoch_output:
@@ -556,7 +556,7 @@ def __auto_reduce_results_on_epoch_end(self, epoch_output, device: Optional[Unio
                 continue
 
             # reduce across training steps
-            opt_outputs = time_reduced_outputs[0].__class__.reduce_on_epoch_end(time_reduced_outputs, device=device)
+            opt_outputs = time_reduced_outputs[0].__class__.reduce_on_epoch_end(time_reduced_outputs)
 
             # with manual opt need 1 + metrics because meta is always there
             if opt_outputs.minimize is not None: