Set default callbacks to None

Signed-off-by: Michal Futrega <michal.futrega@gmail.com>
NVIDIA · Dec 14, 2023 · a1d1471 · a1d1471
1 parent c748a8c
commit a1d1471
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nemo/collections/nlp/parts/megatron_trainer_builder.py b/nemo/collections/nlp/parts/megatron_trainer_builder.py
@@ -95,9 +95,11 @@ def _plugins(self) -> list:
 
         return plugins
 
-    def create_trainer(self, callbacks=(CustomProgressBar(),)) -> Trainer:
+    def create_trainer(self, callbacks=None) -> Trainer:
         strategy = self._training_strategy()
         plugins = self._plugins()
+        if callbacks is None:
+            callbacks = [CustomProgressBar()]
         return Trainer(plugins=plugins, strategy=strategy, **self.cfg.trainer, callbacks=callbacks)