update run_train.sh

f03e88c2 · wandongdong · 8c300451 · f03e88c2 · f03e88c2
Showing with 4 addition and 4 deletion

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

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

未找到文件。
--- a/example/mobilenetv2_quant/scripts/run_train.sh
+++ b/example/mobilenetv2_quant/scripts/run_train.sh
@@ -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


--- a/example/resnet50_quant/scripts/run_train.sh
+++ b/example/resnet50_quant/scripts/run_train.sh
@@ -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