Model tests executable fix

microsoft · Feb 4, 2020 · caaae99 · caaae99
1 parent e9b097f
commit caaae99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/model/Megatron_GPT2/ds_gpt2_test.sh b/tests/model/Megatron_GPT2/ds_gpt2_test.sh
@@ -93,7 +93,7 @@ gpt_options=" \
 
 work_dir="../../../DeepSpeedExamples/Megatron-LM/"
 include_str=`seq 0 $(( $gpus - 1 )) | paste -sd "," -`
-run_cmd="(cd ${work_dir} && deepspeed.pt -i localhost:${include_str} pretrain_gpt2.py ${gpt_options})"
+run_cmd="(cd ${work_dir} && deepspeed -i localhost:${include_str} pretrain_gpt2.py ${gpt_options})"
 echo ${run_cmd}
 eval ${run_cmd}