fix (#6842)

Signed-off-by: Yi Dong <[email protected]>
NVIDIA · Jun 29, 2023 · dbd932f · dbd932f
1 parent 7513fe9
commit dbd932f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb b/tutorials/nlp/Megatron_Synthetic_Tabular_Data_Generation.ipynb
@@ -602,7 +602,7 @@
             "outputs": [],
             "source": [
                 "CHECKPONT_FILE_NAME = megatron_gpt--val_loss=1.17-step=10047-consumed_samples=80376.0-last.ckpt # change it to your checkpoint file name\n",
-                "!python -m torch.distributed.launch --nproc_per_node=1 --use-env=True megatron_ckpt_to_nemo.py \\\n",
+                "!torchrun --nproc_per_node=1 megatron_ckpt_to_nemo.py \\\n",
                 "    --checkpoint_folder=gpt_creditcard_results/megatron_gpt/checkpoints/ \\\n",
                 "    --checkpoint_name={CHECKPONT_FILE_NAME} \\\n",
                 "    --nemo_file_path=tabular.nemo \\\n",