提交 aa796f9d 编写于 作者: Y yejianwu

update run mode in mace_tools.py

上级 30418e13
...@@ -56,3 +56,8 @@ fi ...@@ -56,3 +56,8 @@ fi
if [ "$PRODUCTION_MODE" = 1 ]; then if [ "$PRODUCTION_MODE" = 1 ]; then
cp $GENERATED_MODEL_LIB_PATH $MODEL_OUTPUT_DIR/libmace_${MODEL_TAG}.a cp $GENERATED_MODEL_LIB_PATH $MODEL_OUTPUT_DIR/libmace_${MODEL_TAG}.a
fi fi
if [ -f "$MODEL_OUTPUT_DIR/mace_run" ]; then
rm -rf $MODEL_OUTPUT_DIR/mace_run
fi
cp bazel-bin/examples/mace_run $MODEL_OUTPUT_DIR
...@@ -251,9 +251,6 @@ def main(unused_args): ...@@ -251,9 +251,6 @@ def main(unused_args):
libmace_name = get_libs(configs) libmace_name = get_libs(configs)
if FLAGS.mode == "run" and len(configs) > 1:
raise Exception("Mode 'run' only can execute one model config, which have been built lastest")
model_output_dirs = [] model_output_dirs = []
for config in configs: for config in configs:
# Transfer params by environment # Transfer params by environment
......
...@@ -19,7 +19,7 @@ PRODUCTION_MODE=$4 ...@@ -19,7 +19,7 @@ PRODUCTION_MODE=$4
if [ x"$RUNTIME" = x"local" ]; then if [ x"$RUNTIME" = x"local" ]; then
MACE_CPP_MIN_VLOG_LEVEL=$VLOG_LEVEL \ MACE_CPP_MIN_VLOG_LEVEL=$VLOG_LEVEL \
bazel-bin/examples/mace_run \ ${MODEL_OUTPUT_DIR}/mace_run \
--input_shape="${INPUT_SHAPE}"\ --input_shape="${INPUT_SHAPE}"\
--output_shape="${OUTPUT_SHAPE}"\ --output_shape="${OUTPUT_SHAPE}"\
--input_file=${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} \ --input_file=${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} \
...@@ -38,7 +38,7 @@ else ...@@ -38,7 +38,7 @@ else
adb shell "mkdir -p ${KERNEL_DIR}" || exit 1 adb shell "mkdir -p ${KERNEL_DIR}" || exit 1
fi fi
adb push ${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} ${PHONE_DATA_DIR} || exit 1 adb push ${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} ${PHONE_DATA_DIR} || exit 1
adb push bazel-bin/examples/mace_run ${PHONE_DATA_DIR} || exit 1 adb push ${MODEL_OUTPUT_DIR}/mace_run ${PHONE_DATA_DIR} || exit 1
adb push lib/hexagon/libhexagon_controller.so ${PHONE_DATA_DIR} || exit 1 adb push lib/hexagon/libhexagon_controller.so ${PHONE_DATA_DIR} || exit 1
adb </dev/null shell \ adb </dev/null shell \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册