提交 0e66eb1c 编写于 作者: Y yejianwu

update target_abi to target_abis in config

上级 6acbb9c1
# example.yaml # example.yaml
# Each yaml file describes a exported library (could be named [target_abi]/libmace-${filename}.a), # Each yaml file describes a exported library (could be named [target_abi]/libmace-${filename}.a),
# which can contains more than one models # which can contains more than one models
target_abi: armeabi-v7a # arm64-v8a target_abis: [armeabi-v7a, arm64-v8a]
target_socs: [MSM8953] # target_socs not enabled yet
embed_model_data: 1 embed_model_data: 1
vlog_level: 0 vlog_level: 0
models: models:
......
...@@ -186,14 +186,18 @@ def main(unused_args): ...@@ -186,14 +186,18 @@ def main(unused_args):
if FLAGS.mode == "validate": if FLAGS.mode == "validate":
FLAGS.round = 1 FLAGS.round = 1
target_abi = configs["target_abi"] # target_abi = configs["target_abi"]
libmace_name = get_libs(target_abi, configs) # libmace_name = get_libs(target_abi, configs)
# Transfer params by environment # Transfer params by environment
os.environ["TARGET_ABI"] = target_abi # os.environ["TARGET_ABI"] = target_abi
os.environ["EMBED_MODEL_DATA"] = str(configs["embed_model_data"]) os.environ["EMBED_MODEL_DATA"] = str(configs["embed_model_data"])
os.environ["VLOG_LEVEL"] = str(configs["vlog_level"]) os.environ["VLOG_LEVEL"] = str(configs["vlog_level"])
os.environ["PROJECT_NAME"] = os.path.splitext(FLAGS.config)[0] os.environ["PROJECT_NAME"] = os.path.splitext(FLAGS.config)[0]
for target_abi in configs["target_abis"]:
libmace_name = get_libs(target_abi, configs)
# Transfer params by environment
os.environ["TARGET_ABI"] = target_abi
model_output_dirs = [] model_output_dirs = []
for model_name in configs["models"]: for model_name in configs["models"]:
# Transfer params by environment # Transfer params by environment
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册