diff --git a/model_zoo/official/cv/yolov3_darknet53_quant/README.md b/model_zoo/official/cv/yolov3_darknet53_quant/README.md index 55942c49d6039b62b6a604ceb93632def4b60354..5cd863f876cf7303f63b68710c82e25090170d84 100644 --- a/model_zoo/official/cv/yolov3_darknet53_quant/README.md +++ b/model_zoo/official/cv/yolov3_darknet53_quant/README.md @@ -55,7 +55,7 @@ This is an example of training YOLOV3-DarkNet53-Quant with COCO2014 dataset in M ``` # distributed training -sh run_distribute_train.sh [DATASET_PATH] [RESUME_YOLOV3] [MINDSPORE_HCCL_CONFIG_PATH] +sh run_distribute_train.sh [DATASET_PATH] [RESUME_YOLOV3] [RANK_TABLE_FILE] # standalone training sh run_standalone_train.sh [DATASET_PATH] [RESUME_YOLOV3] diff --git a/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_distribute_train.sh b/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_distribute_train.sh index ced86426327139e3018c2d8cb955f93ee97bdeee..fe024c8433923d439776d2775bbdb32801db6053 100644 --- a/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_distribute_train.sh +++ b/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_distribute_train.sh @@ -16,7 +16,7 @@ if [ $# != 3 ] then - echo "Usage: sh run_distribute_train.sh [DATASET_PATH] [RESUME_YOLOV3] [MINDSPORE_HCCL_CONFIG_PATH]" + echo "Usage: sh run_distribute_train.sh [DATASET_PATH] [RESUME_YOLOV3] [RANK_TABLE_FILE]" exit 1 fi @@ -30,11 +30,11 @@ get_real_path(){ DATASET_PATH=$(get_real_path $1) RESUME_YOLOV3=$(get_real_path $2) -MINDSPORE_HCCL_CONFIG_PATH=$(get_real_path $3) +RANK_TABLE_FILE=$(get_real_path $3) echo $DATASET_PATH echo $RESUME_YOLOV3 -echo $MINDSPORE_HCCL_CONFIG_PATH +echo $RANK_TABLE_FILE if [ ! -d $DATASET_PATH ] then @@ -48,15 +48,15 @@ then exit 1 fi -if [ ! -f $MINDSPORE_HCCL_CONFIG_PATH ] +if [ ! -f $RANK_TABLE_FILE ] then - echo "error: MINDSPORE_HCCL_CONFIG_PATH=$MINDSPORE_HCCL_CONFIG_PATH is not a file" + echo "error: RANK_TABLE_FILE=$RANK_TABLE_FILE is not a file" exit 1 fi export DEVICE_NUM=8 export RANK_SIZE=8 -export MINDSPORE_HCCL_CONFIG_PATH=$MINDSPORE_HCCL_CONFIG_PATH +export RANK_TABLE_FILE=$RANK_TABLE_FILE for((i=0; i<${DEVICE_NUM}; i++)) do