提交 aa796f9d 编写于 作者: Y yejianwu

update run mode in mace_tools.py

上级 30418e13
......@@ -56,3 +56,8 @@ fi
if [ "$PRODUCTION_MODE" = 1 ]; then
cp $GENERATED_MODEL_LIB_PATH $MODEL_OUTPUT_DIR/libmace_${MODEL_TAG}.a
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):
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 = []
for config in configs:
# Transfer params by environment
......
......@@ -19,7 +19,7 @@ PRODUCTION_MODE=$4
if [ x"$RUNTIME" = x"local" ]; then
MACE_CPP_MIN_VLOG_LEVEL=$VLOG_LEVEL \
bazel-bin/examples/mace_run \
${MODEL_OUTPUT_DIR}/mace_run \
--input_shape="${INPUT_SHAPE}"\
--output_shape="${OUTPUT_SHAPE}"\
--input_file=${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} \
......@@ -38,7 +38,7 @@ else
adb shell "mkdir -p ${KERNEL_DIR}" || exit 1
fi
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 </dev/null shell \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册