提交 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):
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):
......
......@@ -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 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
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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册