Browse Source

replace MINDSPORE_HCCL_CONFIG in yolov3_darknet53_quant

tags/v0.7.0-beta
panbingao 5 years ago
parent
commit
de0e1f00ce
2 changed files with 7 additions and 7 deletions
  1. +1
    -1
      model_zoo/official/cv/yolov3_darknet53_quant/README.md
  2. +6
    -6
      model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_distribute_train.sh

+ 1
- 1
model_zoo/official/cv/yolov3_darknet53_quant/README.md View File

@@ -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]


+ 6
- 6
model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_distribute_train.sh View File

@@ -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


Loading…
Cancel
Save