From 145f297c8d79c94ab5bffdf51a908656a4585fce Mon Sep 17 00:00:00 2001 From: linqingke Date: Mon, 12 Jul 2021 16:59:43 +0800 Subject: [PATCH] fix multi server accuracy bug. --- .../official/cv/resnet/scripts/run_distribute_train.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/model_zoo/official/cv/resnet/scripts/run_distribute_train.sh b/model_zoo/official/cv/resnet/scripts/run_distribute_train.sh index 017c3cffcd3..0d479c41863 100755 --- a/model_zoo/official/cv/resnet/scripts/run_distribute_train.sh +++ b/model_zoo/official/cv/resnet/scripts/run_distribute_train.sh @@ -107,19 +107,19 @@ do env > env.log if [ $# == 3 ] then - taskset -c $cmdopt python train.py --run_distribute=True --device_num=$DEVICE_NUM --data_path=$PATH2 \ + taskset -c $cmdopt python train.py --run_distribute=True --device_num=$RANK_SIZE --data_path=$PATH2 \ --config_path=$CONFIG_FILE --output_path './output' &> log & fi if [ $# == 4 ] then - taskset -c $cmdopt python train.py --run_distribute=True --device_num=$DEVICE_NUM --data_path=$PATH2 --pre_trained=$PATH3 \ + taskset -c $cmdopt python train.py --run_distribute=True --device_num=$RANK_SIZE --data_path=$PATH2 --pre_trained=$PATH3 \ --config_path=$CONFIG_FILE --output_path './output' &> log & fi if [ $# == 5 ] then - taskset -c $cmdopt python train.py --run_distribute=True --device_num=$DEVICE_NUM --data_path=$PATH2 \ + taskset -c $cmdopt python train.py --run_distribute=True --device_num=$RANK_SIZE --data_path=$PATH2 \ --run_eval=$RUN_EVAL --eval_data_path=$EVAL_DATASET_PATH --enable_cache=True --cache_session_id=$CACHE_SESSION_ID \ --config_path=$CONFIG_FILE --output_path './output' &> log & if [ "x${RUN_EVAL}" == "xTrue" ]