p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug for ssd

This commit is contained in:
chengxianbin 2020-06-11 11:55:11 +08:00
parent ffb5339e87
commit b92fed27a8
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
model_zoo/ssd/scripts/run_distribute_train.sh
View File

@ -29,6 +29,8 @@ then
fi fi
# Before start distribute train, first create mindrecord files. # Before start distribute train, first create mindrecord files.
BASE_PATH=$(cd "`dirname $0`" || exit; pwd)
cd $BASE_PATH/../ || exit
python train.py --only_create_dataset=1 python train.py --only_create_dataset=1
echo "After running the scipt, the network runs in the background. The log will be generated in LOGx/log.txt" echo "After running the scipt, the network runs in the background. The log will be generated in LOGx/log.txt"
@ -46,8 +48,8 @@ do
export DEVICE_ID=$i export DEVICE_ID=$i
rm -rf LOG$i rm -rf LOG$i
mkdir ./LOG$i mkdir ./LOG$i
cp ../*.py ./LOG$i cp ./*.py ./LOG$i
cp -r ../src ./LOG$i cp -r ./src ./LOG$i
cd ./LOG$i || exit cd ./LOG$i || exit
export RANK_ID=$i export RANK_ID=$i
echo "start training for rank $i, device $DEVICE_ID" echo "start training for rank $i, device $DEVICE_ID"