fix bug

argonne-lcf · Feb 14, 2025 · 3a1cbc0 · 3a1cbc0
1 parent 70e889b
commit 3a1cbc0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dlio_benchmark/main.py b/dlio_benchmark/main.py
@@ -251,8 +251,8 @@ def _train(self, epoch):
         self.stats.start_loading()
         for batch in loader.next():
             if overall_step > max_steps or ((self.total_training_steps > 0) and (overall_step > self.total_training_steps)):
-            if self.args.my_rank == 0:
-                logging.info(f"{utcnow()} Maximum number of steps reached")
+                if self.args.my_rank == 0:
+                    logging.info(f"{utcnow()} Maximum number of steps reached")
             if (block_step != 1 and self.do_checkpoint) or (not self.do_checkpoint):
                 self.stats.end_block(epoch, block, block_step - 1)
                 break