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

!4457 modify yolov3_quant eval script 

Merge pull request !4457 from chengxb7532/r0.6
This commit is contained in:
mindspore-ci-bot 2020-08-15 10:33:40 +08:00 committed by Gitee
parent 04a6612baf ef9e3a5360
commit 50d7480a4e
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_eval.sh
View File

@ -14,9 +14,9 @@
# limitations under the License.
# ============================================================================
if [ $# != 2 ]
if [ $# != 3 ]
then
echo "Usage: sh run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH]"
echo "Usage: sh run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH] [DEVICE_ID]"
exit 1
fi
@ -45,19 +45,19 @@ exit 1
fi
export DEVICE_NUM=1
export DEVICE_ID=0
export DEVICE_ID=$3
export RANK_SIZE=$DEVICE_NUM
export RANK_ID=0
if [ -d "eval" ];
if [ -d "eval$3" ];
then
rm -rf ./eval
rm -rf ./eval$3
fi
mkdir ./eval
cp ../*.py ./eval
cp -r ../src ./eval
cd ./eval || exit
mkdir ./eval$3
cp ../*.py ./eval$3
cp -r ../src ./eval$3
cd ./eval$3 || exit
env > env.log
echo "start infering for device $DEVICE_ID"
python eval.py \