Add '32-true' for precision values

Signed-off-by: Abhishree Thittenamane <[email protected]>
NVIDIA · Oct 20, 2023 · 9a545fb · 9a545fb
1 parent 202cbcb
commit 9a545fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/nlp/language_modeling/megatron_ckpt_to_nemo.py b/examples/nlp/language_modeling/megatron_ckpt_to_nemo.py
@@ -102,7 +102,7 @@ def get_args():
         type=str,
         required=False,
         default='16-mixed',
-        choices=['16-mixed', 'bf16-mixed'],
+        choices=['32-true', '16-mixed', 'bf16-mixed'],
         help="Precision value for the trainer that matches with precision of the ckpt",
     )