From b92fed27a8e99da31c485c71cdbb1d3fc3c37b94 Mon Sep 17 00:00:00 2001 From: chengxianbin Date: Thu, 11 Jun 2020 11:55:11 +0800 Subject: [PATCH] fix bug for ssd --- model_zoo/ssd/scripts/run_distribute_train.sh | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/model_zoo/ssd/scripts/run_distribute_train.sh b/model_zoo/ssd/scripts/run_distribute_train.sh index accd35c50c7..60eccf2c40c 100644 --- a/model_zoo/ssd/scripts/run_distribute_train.sh +++ b/model_zoo/ssd/scripts/run_distribute_train.sh @@ -29,6 +29,8 @@ then fi # 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 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 rm -rf LOG$i mkdir ./LOG$i - cp ../*.py ./LOG$i - cp -r ../src ./LOG$i + cp ./*.py ./LOG$i + cp -r ./src ./LOG$i cd ./LOG$i || exit export RANK_ID=$i echo "start training for rank $i, device $DEVICE_ID"