removed unnecessary print (#6884)

Signed-off-by: Dmytro Pykhtar <dpykhtar@nvidia.com>
NVIDIA · Jun 21, 2023 · 24837af · 24837af
1 parent 63d9b2c
commit 24837af
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nemo/collections/nlp/models/language_modeling/megatron_gpt_model.py b/nemo/collections/nlp/models/language_modeling/megatron_gpt_model.py
@@ -518,7 +518,6 @@ def training_step(self, dataloader_iter, batch_idx):
         if self.rampup_batch_size:
             num_microbatch_calculator = apex.transformer.pipeline_parallel.utils._GLOBAL_NUM_MICROBATCHES_CALCULATOR
             current_global_batch_size = num_microbatch_calculator.current_global_batch_size
-            logging.info(current_global_batch_size)
             # do validation and save the checkpoint when gbs is changed
             if self.prev_global_batch_size != current_global_batch_size and self.prev_global_batch_size:
                 self.trainer.should_stop = True