diff --git a/validate.py b/validate.py index 289ec30853aad004970b541d436a0a005a8656f1..4978a995d96bad226f288bb89a348d826153ecd4 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 b400faa50d3eba0a321468e32a22d9e96756adf3..78dfce7550ee7658ac78a3f2c2ee84d029f810ad 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