debug

Signed-off-by: Yang Zhang <yangzhang@nvidia.com>
NVIDIA · Feb 11, 2020 · 9b5acd4 · 9b5acd4
1 parent 8bf4389
commit 9b5acd4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/nlp/language_modeling/bert_pretraining.py b/examples/nlp/language_modeling/bert_pretraining.py
@@ -84,6 +84,8 @@
 from nemo.collections.nlp.data.datasets.lm_bert_dataset import BERTPretrainingDataDesc
 from nemo.utils.lr_policies import get_lr_policy
 
+print("nemo_nlp path", nemo_nlp.__file__)
+
 parser = argparse.ArgumentParser(description='BERT pretraining')
 parser.add_argument("--local_rank", default=None, type=int)
 parser.add_argument("--num_gpus", default=1, type=int)