提交 7a0e8cf0 编写于 作者: Y yejianwu

remove unnecessary files, rename model.config, use relative dir to libmace root in config

create build/lib/libmace_generated.a
addlib lib/mace/libmace.a
addlib lib/mace/libmace_prod.a
addlib bazel-bin/codegen/libgenerated_models.a
addlib bazel-bin/codegen/libgenerated_opencl_prod.a
addlib bazel-bin/codegen/libgenerated_tuning_params.a
save
end
create build/lib/libmace_generated.a
addlib lib/mace/libmace.a
addlib lib/mace/libmace_prod.a
addlib bazel-bin/codegen/libgenerated_models.pic.a
save
end
#!/usr/bin/env python
# Must run at root dir of libmace project.
# python tools/mace_tools.py \
# --global_config=models/config \
# --round=100 \
......@@ -178,21 +179,18 @@ def parse_model_configs():
model_dirs = []
model_output_map = {}
if ("models" in config_sections) and (config_parser.items("models")):
model_dirs_relative_to_config_str = config_parser.get(
model_dirs_str = config_parser.get(
"models", "DIRECTORIES")
model_dirs_relative_to_config_str = model_dirs_relative_to_config_str.rstrip(
model_dirs_str = model_dirs_str.rstrip(
",")
# Remove repetition element
model_dirs_relative_to_config_str_list = list(
set(model_dirs_relative_to_config_str.split(",")))
for d in model_dirs_relative_to_config_str_list:
model_dir = os.path.join(global_config_dir, d)
model_dirs.append(model_dir)
model_dirs = list(
set(model_dirs_str.split(",")))
for model_dir in model_dirs:
# Create output dirs
model_output_dir = os.path.join(FLAGS.output_dir, d)
model_output_dir = os.path.join(FLAGS.output_dir, model_dir)
model_output_map[model_dir] = model_output_dir
else:
......
; [models]
; DIRECTORIES = mobile_net/64/
[configs]
; If exists 'models' section, this section will be global variable to sub models
; Otherwise this is the model's config and must be put in the same dirs with model.pb
TF_INPUT_NODE=input
TF_OUTPUT_NODE=output
MODEL_TAG=mobile_net
INPUT_SHAPE=1,64,64,3
OUTPUT_SHAPE=1,64,64,2
RUNTIME=gpu
TUNING_OR_NOT=1
ANDROID_ABI=armeabi-v7a ; Or 'arm64-v8a'
LIMIT_OPENCL_KERNEL_TIME=0
DSP_MODE=0 ; used only for dsp
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册