bugfix (NVIDIA#10561)

Signed-off-by: Maanu Grover <maanug@nvidia.com>
JimmyZhang12 · Sep 23, 2024 · 9ed0d6c · 9ed0d6c
1 parent 110db0c
commit 9ed0d6c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nemo/lightning/pytorch/callbacks/model_checkpoint.py b/nemo/lightning/pytorch/callbacks/model_checkpoint.py
@@ -163,6 +163,8 @@ def on_train_start(self, trainer, pl_module):
         if torch.distributed.is_initialized():
             torch.distributed.barrier()
 
+        super().on_train_start(trainer, pl_module)
+
     def nemo_topk_check_previous_run(self):
         try:
             self.best_k_models