Model tests executable fix

caaae992 · Shaden Smith · GitHub · e9b097f1 · caaae992
Unverified Commit caaae992 authored Feb 03, 2020 by Shaden Smith Committed by GitHub Feb 03, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/model/Megatron_GPT2/ds_gpt2_test.sh tests/model/Megatron_GPT2/ds_gpt2_test.sh +1 -1

No files found.
--- 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}