提交 d536f99f 编写于 作者: T tangwei

remove unused flag -d -e

上级 3cae956f
...@@ -33,7 +33,9 @@ model_name = "" ...@@ -33,7 +33,9 @@ model_name = ""
def engine_registry(): def engine_registry():
engines = {"TRANSPILER": {}, "PSLIB": {}} engines["TRANSPILER"] = {}
engines["PSLIB"] = {}
engines["TRANSPILER"]["SINGLE"] = single_engine engines["TRANSPILER"]["SINGLE"] = single_engine
engines["TRANSPILER"]["LOCAL_CLUSTER"] = local_cluster_engine engines["TRANSPILER"]["LOCAL_CLUSTER"] = local_cluster_engine
engines["TRANSPILER"]["CLUSTER"] = cluster_engine engines["TRANSPILER"]["CLUSTER"] = cluster_engine
...@@ -60,13 +62,16 @@ def get_engine(args): ...@@ -60,13 +62,16 @@ def get_engine(args):
transpiler = get_transpiler() transpiler = get_transpiler()
run_extras = get_inters_from_yaml(args.model, "train.") run_extras = get_inters_from_yaml(args.model, "train.")
engine = run_extras.get("train.engine", "") engine = run_extras.get("train.engine", "single")
engine = engine.upper() engine = engine.upper()
if engine not in engine_choices: if engine not in engine_choices:
raise ValueError("train.engin can not be chosen in {}".format(engine_choices)) raise ValueError("train.engin can not be chosen in {}".format(engine_choices))
print("engines: \n{}".format(engines))
run_engine = engines[transpiler].get(engine, None) run_engine = engines[transpiler].get(engine, None)
return run_engine return run_engine
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册