examples -> DeepSpeedExamples

e9b097f1 · Shaden Smith · 681001fb · e9b097f1
Commit e9b097f1 authored Feb 03, 2020 by Shaden Smith
Hide 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
@@ -91,7 +91,7 @@ gpt_options=" \
       ${zero_opt} \
 "

-work_dir="../../../examples/Megatron-LM/"
+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})"
 echo ${run_cmd}