From d536f99f51e07e6643a547dec982160fb3a6bd49 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Mon, 18 May 2020 15:58:33 +0800
Subject: [PATCH] remove unused flag -d -e

---
 run.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/run.py b/run.py
index 94a9386f..41928089 100755
--- a/run.py
+++ b/run.py
@@ -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
 
 
-- 
GitLab