提交 d536f99f 编写于 作者: T tangwei

remove unused flag -d -e

上级 3cae956f
......@@ -33,7 +33,9 @@ model_name = ""
def engine_registry():
engines = {"TRANSPILER": {}, "PSLIB": {}}
engines["TRANSPILER"] = {}
engines["PSLIB"] = {}
engines["TRANSPILER"]["SINGLE"] = single_engine
engines["TRANSPILER"]["LOCAL_CLUSTER"] = local_cluster_engine
engines["TRANSPILER"]["CLUSTER"] = cluster_engine
......@@ -60,13 +62,16 @@ def get_engine(args):
transpiler = get_transpiler()
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()
if engine not in 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)
return run_engine
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册