From 517b0d4c449366717ed041ffe0c5ec9a1d17958c Mon Sep 17 00:00:00 2001 From: ashors1 Date: Tue, 7 Jan 2025 09:42:31 -0800 Subject: [PATCH] make progress printer compatible with PTL v2.5.0 Signed-off-by: ashors1 --- nemo/lightning/pytorch/callbacks/progress_printer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/lightning/pytorch/callbacks/progress_printer.py b/nemo/lightning/pytorch/callbacks/progress_printer.py index 12d05ed2950c..5aabcb88f976 100644 --- a/nemo/lightning/pytorch/callbacks/progress_printer.py +++ b/nemo/lightning/pytorch/callbacks/progress_printer.py @@ -111,7 +111,7 @@ def on_sanity_check_end(self, *_: Any) -> None: self._validation_description = "Validation" @override - def on_train_epoch_start(self, trainer, *_): + def on_train_start(self, trainer, *_): if trainer.max_steps > 0: # while resuming from a ckpt use trainer.max_steps as the total for progress bar as trainer.num_training_batches # is truncated to max_steps - step being resumed at