nvidia
/

nemo-megatron-t5-3B

masked language modeling

Model card Files Files and versions Community

MaximumEntropy commited on Sep 21, 2022

Commit

d43eded

•

1 Parent(s): 002adce

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ git clone https://github.com/NVIDIA/NeMo.git
 cd NeMo/examples/nlp/language_modeling
 git checkout v1.11.0
 python megatron_t5_eval.py \
-    --model_file /raid/Data/NMT/Models/t5_3b/megatron_t5-tp2--val_los-1.09-step-999999-consumed-samples-2159846144.0.nemo \
     --prompt '<mask> was the first person to set foot on the moon. When he did, he uttered the phrase <mask> for man, one <mask> for mankind which is still a popular quote today.' \
     --tensor_model_parallel_size 2
 ```

 cd NeMo/examples/nlp/language_modeling
 git checkout v1.11.0
 python megatron_t5_eval.py \
+    --model_file /raid/Data/NMT/Models/t5_3b/nemo_megatron_t5_3b_bf16_tp2.nemo \
     --prompt '<mask> was the first person to set foot on the moon. When he did, he uttered the phrase <mask> for man, one <mask> for mankind which is still a popular quote today.' \
     --tensor_model_parallel_size 2
 ```