start training loop (#3005)

mosaicml · Feb 13, 2024 · 953ee88 · 953ee88
1 parent 4205425
commit 953ee88
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py
@@ -2086,6 +2086,7 @@ def _train_loop(self) -> None:
             # asserted to be not None when Trainer.fit() is called
             raise RuntimeError('max_duration must be specified when initializing the Trainer')
 
+        log.debug('Starting training loop')
         while self.state.timestamp < self.state.max_duration:
             if int(self.state.timestamp.batch_in_epoch) == 0:
                 self.engine.run_event(Event.EPOCH_START)