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

!19902 resnet dataset bug fix master 

Merge pull request !19902 from 张晓晓/code_docs_resnet
This commit is contained in:
i-robot 2021-07-10 06:42:54 +00:00 committed by Gitee
parent 8ab30e0638 a502aae49c
commit 09a119cd7c
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
model_zoo/official/cv/resnet/scripts/run_infer_310.sh
View File

@ -17,6 +17,7 @@
if [[ $# -lt 4 || $# -gt 5 ]]; then
echo "Usage: bash run_infer_310.sh [MINDIR_PATH] [NET_TYPE] [DATASET] [DATA_PATH] [DEVICE_ID]
NET_TYPE can choose from [resnet18, resnet34, se-resnet50, resnet50, resnet101]
DATASET can choose from [cifar10, imagenet]
DEVICE_ID is optional, it can be set by environment variable device_id, otherwise the value is zero"
exit 1
fi
@ -36,7 +37,13 @@ else
exit 1
fi
dataset=$3
if [ $3 == 'cifar10' ] || [ $3 == 'imagenet' ]; then
dataset=$3
else
echo "DATASET can choose from [cifar10, imagenet]"
exit 1
fi
data_path=$(get_real_path $4)
device_id=0