unused import

Signed-off-by: Maanu Grover <[email protected]>
NVIDIA · Aug 3, 2023 · 3408fc8 · 3408fc8
1 parent ee8accb
commit 3408fc8
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/examples/nlp/language_modeling/megatron_bert_pretraining.py b/examples/nlp/language_modeling/megatron_bert_pretraining.py
@@ -15,7 +15,6 @@
 import torch.multiprocessing as mp
 from megatron_trainer_builder import MegatronBertTrainerBuilder
 from omegaconf.omegaconf import OmegaConf, open_dict
-from pytorch_lightning.trainer.connectors.checkpoint_connector import CheckpointConnector
 
 from nemo.collections.nlp.models.language_modeling.megatron_bert_model import MegatronBertModel
 from nemo.core.config import hydra_runner

diff --git a/examples/nlp/language_modeling/megatron_t5_pretraining.py b/examples/nlp/language_modeling/megatron_t5_pretraining.py
@@ -15,7 +15,6 @@
 
 from megatron_trainer_builder import MegatronT5TrainerBuilder
 from omegaconf.omegaconf import OmegaConf, open_dict
-from pytorch_lightning.trainer.connectors.checkpoint_connector import CheckpointConnector
 
 from nemo.collections.nlp.models.language_modeling.megatron_t5_model import MegatronT5Model
 from nemo.core.config import hydra_runner