logging with fdr

Lightning-AI · Nov 15, 2020 · e098abe · e098abe
1 parent 463dad8
commit e098abe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pytorch_lightning/trainer/connectors/debugging_connector.py b/pytorch_lightning/trainer/connectors/debugging_connector.py
@@ -44,9 +44,7 @@ def on_init_start(
             )
 
         self.trainer.fast_dev_run = fast_dev_run
-
-        if fast_dev_run is True:
-            fast_dev_run = 1
+        fast_dev_run = int(fast_dev_run)
 
         if fast_dev_run:
             limit_train_batches = fast_dev_run

diff --git a/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py b/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py
@@ -613,7 +613,7 @@ def __gather_result_across_time_and_optimizers(self, epoch_output):
 
     def log_train_step_metrics(self, batch_output):
         # when metrics should be logged
-        if self.should_update_logs or self.trainer.fast_dev_run:
+        if self.should_update_logs or (int(self.trainer.fast_dev_run) == 1):
             # logs user requested information to logger
             metrics = self.cached_results.get_latest_batch_log_metrics()
             grad_norm_dic = batch_output.grad_norm_dic