diff --git a/research/cv/FaceAttribute/scripts/run_distribute_train.sh b/research/cv/FaceAttribute/scripts/run_distribute_train.sh index c4176604d300c40eeb3969f6467980c193ec5c55..4a324d69171d82ce195641ca207bfe749ef31052 100644 --- a/research/cv/FaceAttribute/scripts/run_distribute_train.sh +++ b/research/cv/FaceAttribute/scripts/run_distribute_train.sh @@ -64,16 +64,24 @@ echo $PRETRAINED_BACKBONE export RANK_TABLE_FILE=$RANK_TABLE export RANK_SIZE=8 +cpus=`cat /proc/cpuinfo| grep "processor"| wc -l` +avg=`expr $cpus \/ $RANK_SIZE` +gap=`expr $avg \- 1` + echo 'start training' for((i=0;i<=$RANK_SIZE-1;i++)); do + start=`expr $i \* $avg` + end=`expr $start \+ $gap` + cmdopt=$start"-"$end + echo 'start rank '$i mkdir ${current_exec_path}/device$i cd ${current_exec_path}/device$i || exit export RANK_ID=$i dev=`expr $i + 0` export DEVICE_ID=$dev - python ${dirname_path}/${SCRIPT_NAME} \ + taskset -c $cmdopt python ${dirname_path}/${SCRIPT_NAME} \ --mindrecord_path=$MINDRECORD_FILE \ --pretrained=$PRETRAINED_BACKBONE > train.log 2>&1 & done