diff --git a/cifar/run_ds.sh b/cifar/run_ds.sh index 888f25c57..7fcce356b 100755 --- a/cifar/run_ds.sh +++ b/cifar/run_ds.sh @@ -1,9 +1,3 @@ #!/bin/bash -num_gpus=1 -ds_config=ds_config.json - -args="--deepspeed --deepspeed_config ${ds_config}" -run_cmd="deepspeed --num_nodes 1 --num_gpus ${num_gpus} cifar10_deepspeed.py $@ ${args}" -echo ${run_cmd} -eval ${run_cmd} +deepspeed cifar10_deepspeed.py --deepspeed --deepspeed_config ds_config.json