Browse Source

update run_train.sh

tags/v0.3.1-alpha
wandongdong 6 years ago
parent
commit
f03e88c26f
2 changed files with 4 additions and 4 deletions
  1. +2
    -2
      example/mobilenetv2_quant/scripts/run_train.sh
  2. +2
    -2
      example/resnet50_quant/scripts/run_train.sh

+ 2
- 2
example/mobilenetv2_quant/scripts/run_train.sh View File

@@ -22,9 +22,9 @@ run_ascend()
exit 1
fi

if [ ! -d $5 ]
if [ ! -d $5 ] && [ ! -f $5 ]
then
echo "error: DATASET_PATH=$5 is not a directory"
echo "error: DATASET_PATH=$5 is not a directory or file"
exit 1
fi



+ 2
- 2
example/resnet50_quant/scripts/run_train.sh View File

@@ -22,9 +22,9 @@ run_ascend()
exit 1
fi

if [ ! -d $5 ]
if [ ! -d $5 ] && [ ! -f $5 ]
then
echo "error: DATASET_PATH=$5 is not a directory"
echo "error: DATASET_PATH=$5 is not a directory or file"
exit 1
fi



Loading…
Cancel
Save