Fix logging interval (#2694)

Lightning-AI · Jul 25, 2020 · 8599b67 · 8599b67
1 parent 4794175
commit 8599b67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py
@@ -632,7 +632,7 @@ def increment_accumulated_grad_global_step(self):
 
     def save_train_loop_metrics_to_loggers(self, batch_idx, batch_output):
         # when metrics should be logged
-        should_log_metrics = batch_idx % self.row_log_interval == 0 or self.should_stop
+        should_log_metrics = (batch_idx + 1) % self.row_log_interval == 0 or self.should_stop
         if should_log_metrics or self.fast_dev_run:
             # logs user requested information to logger
             metrics = batch_output.batch_log_metrics