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