Commit 1ab8954b authored by panning's avatar panning
Browse files

修改多卡训练脚本路径

parent f7d3c321
Pipeline #164 canceled with stages
...@@ -5,7 +5,7 @@ lrank=$OMPI_COMM_WORLD_LOCAL_RANK ...@@ -5,7 +5,7 @@ lrank=$OMPI_COMM_WORLD_LOCAL_RANK
comm_rank=$OMPI_COMM_WORLD_RANK comm_rank=$OMPI_COMM_WORLD_RANK
comm_size=$OMPI_COMM_WORLD_SIZE comm_size=$OMPI_COMM_WORLD_SIZE
APP="python3 `pwd`/main_acc.py --batch-size=${3} --arch=${2} -j 6 --epochs=90 --dist-url tcp://${1}:34567 --dist-backend nccl --world-size=${comm_size} --rank=${comm_rank} --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/" APP="python3 ../train.py --batch-size=${3} --arch=${2} -j 6 --epochs=90 --dist-url tcp://${1}:34567 --dist-backend nccl --world-size=${comm_size} --rank=${comm_rank} --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/"
case ${lrank} in case ${lrank} in
[0]) [0])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment