提交 b1ce72f9 编写于 作者: L liuqi

Exit program with error code for validation.

上级 894302da
...@@ -38,8 +38,10 @@ def compare_output(output_name, mace_out_value, out_value): ...@@ -38,8 +38,10 @@ def compare_output(output_name, mace_out_value, out_value):
print '=======================Similarity Test Passed======================' print '=======================Similarity Test Passed======================'
else: else:
print '=======================Similarity Test Failed======================' print '=======================Similarity Test Failed======================'
sys.exit(-1)
else: else:
print '=======================Skip empty node===================' print '=======================Skip empty node==================='
sys.exit(-1)
def validate_tf_model(input_names, input_shapes, output_names): def validate_tf_model(input_names, input_shapes, output_names):
......
...@@ -35,7 +35,7 @@ if [ "$PLATFORM" == "tensorflow" ];then ...@@ -35,7 +35,7 @@ if [ "$PLATFORM" == "tensorflow" ];then
for NAME in "${OUTPUT_NAMES[@]}";do for NAME in "${OUTPUT_NAMES[@]}";do
FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME}) FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME})
rm -rf ${MODEL_OUTPUT_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_NAME} rm -rf ${MODEL_OUTPUT_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_NAME}
adb </dev/null 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 done
fi fi
python tools/validate.py --platform=tensorflow \ python tools/validate.py --platform=tensorflow \
...@@ -77,7 +77,7 @@ elif [ "$PLATFORM" == "caffe" ];then ...@@ -77,7 +77,7 @@ elif [ "$PLATFORM" == "caffe" ];then
for NAME in "${OUTPUT_NAMES[@]}";do for NAME in "${OUTPUT_NAMES[@]}";do
FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME}) FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME})
rm -rf ${MODEL_OUTPUT_DIR}/${OUTPUT_FILE_NAME}_${FORMATTED_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 done
fi fi
for NAME in "${OUTPUT_NAMES[@]}";do for NAME in "${OUTPUT_NAMES[@]}";do
...@@ -99,6 +99,6 @@ elif [ "$PLATFORM" == "caffe" ];then ...@@ -99,6 +99,6 @@ elif [ "$PLATFORM" == "caffe" ];then
--input_node ${INPUT_NODES} \ --input_node ${INPUT_NODES} \
--output_node ${OUTPUT_NODES} \ --output_node ${OUTPUT_NODES} \
--input_shape ${INPUT_SHAPES} \ --input_shape ${INPUT_SHAPES} \
--output_shape ${OUTPUT_SHAPES} --output_shape ${OUTPUT_SHAPES} || exit 1
fi fi
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册