From b1ce72f98abbecc9f6b52821163e7127160b7a75 Mon Sep 17 00:00:00 2001 From: liuqi Date: Sat, 10 Mar 2018 11:03:49 +0800 Subject: [PATCH] Exit program with error code for validation. --- validate.py | 2 ++ validate_tools.sh | 6 +++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/validate.py b/validate.py index 289ec308..4978a995 100644 --- a/validate.py +++ b/validate.py @@ -38,8 +38,10 @@ def compare_output(output_name, mace_out_value, out_value): print '=======================Similarity Test Passed======================' else: print '=======================Similarity Test Failed======================' + sys.exit(-1) else: print '=======================Skip empty node===================' + sys.exit(-1) def validate_tf_model(input_names, input_shapes, output_names): diff --git a/validate_tools.sh b/validate_tools.sh index b400faa5..78dfce75 100644 --- a/validate_tools.sh +++ b/validate_tools.sh @@ -35,7 +35,7 @@ if [ "$PLATFORM" == "tensorflow" ];then for NAME in "${OUTPUT_NAMES[@]}";do FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME}) rm -rf ${MODEL_OUTPUT_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_NAME} - adb /dev/null done fi python tools/validate.py --platform=tensorflow \ @@ -77,7 +77,7 @@ elif [ "$PLATFORM" == "caffe" ];then for NAME in "${OUTPUT_NAMES[@]}";do FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME}) rm -rf ${MODEL_OUTPUT_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_NAME} - adb pull ${PHONE_DATA_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_NAME} ${MODEL_OUTPUT_DIR} + adb pull ${PHONE_DATA_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_NAME} ${MODEL_OUTPUT_DIR} > /dev/null done fi for NAME in "${OUTPUT_NAMES[@]}";do @@ -99,6 +99,6 @@ elif [ "$PLATFORM" == "caffe" ];then --input_node ${INPUT_NODES} \ --output_node ${OUTPUT_NODES} \ --input_shape ${INPUT_SHAPES} \ - --output_shape ${OUTPUT_SHAPES} + --output_shape ${OUTPUT_SHAPES} || exit 1 fi -- GitLab