fix readme

fleetrec/core/factory.py

@@ -50,9 +50,8 @@ class TrainerFactory(object):
             trainer_abs = train_mode
             train_mode = "UserDefineTrainer"
 
-        train_location = envs.get_global_env("train.location")
         train_dirname = os.path.dirname(trainer_abs)
-        base_name = os.path.splitext(os.path.basename(train_location))[0]
+        base_name = os.path.splitext(os.path.basename(trainer_abs))[0]
         sys.path.append(train_dirname)
         trainer_class = envs.lazy_instance(base_name, train_mode)
         trainer = trainer_class(yaml_path)

fleetrec/core/trainers/transpiler_trainer.py

@@ -44,7 +44,7 @@ class TranspileTrainer(Trainer):
         reader_class = envs.get_global_env("class", None, namespace)
         abs_dir = os.path.dirname(os.path.abspath(__file__))
         reader = os.path.join(abs_dir, '../utils', 'reader_instance.py')
-        pipe_cmd = "python {} {} {} {}".format(reader, reader_class, "TRAIN", self._config)
+        pipe_cmd = "python {} {} {} {}".format(reader, reader_class, "TRAIN", self._config_yaml)
         train_data_path = envs.get_global_env("train_data_path", None, namespace)
 
         dataset = fluid.DatasetFactory().create_dataset()

fleetrec/run.py

@@ -40,10 +40,8 @@ def get_engine(engine):
 
 def single_engine(args):
     print("use single engine to run model: {}".format(args.model))
-    single_envs = {"trainer.trainer": "SingleTrainer"}
-
+    single_envs = {"trainer.trainer": "SingleTrainer", "trainer.threads": "2"}
     set_runtime_envs(single_envs, args.engine_extras)
-
     trainer = TrainerFactory.create(args.model)
     return trainer