MCPcopy Create free account
hub / github.com/XiaoMi/mace / run_model

Function run_model

tools/python/run_micro.py:67–74  ·  view source on GitHub ↗
(flags, args, conf)

Source from the content-addressed store, hash-verified

65
66
67def run_model(flags, args, conf):
68 model_conf = get_model_conf_by_name(flags, conf)
69 mace_check(model_conf is not None, "Get model conf failed.")
70 model_conf = config_parser.normalize_model_config(model_conf)
71 mace_check(len(model_conf[ModelKeys.subgraphs].items()) == 1,
72 "MACE Micro does not support multi graph.")
73 for graph_name, graph_config in model_conf[ModelKeys.subgraphs].items():
74 run_model_with_conf(flags, args, flags.model_name, graph_config)
75
76
77def gen_sub_model_conf(output_config, flags, conf):

Callers 1

run_micro.pyFile · 0.85

Calls 3

mace_checkFunction · 0.90
get_model_conf_by_nameFunction · 0.85
run_model_with_confFunction · 0.85

Tested by

no test coverage detected