code clean

fleetrec/examples/user_define_trainer.py

+from fleetrec.trainer.trainer import Trainer
+
+
+class UserDefineTrainer(Trainer):
+    def __init__(self, config=None):
+        Trainer.__init__(self, config)

fleetrec/trainer/factory.py

@@ -34,15 +34,22 @@ class TrainerFactory(object):
         print(envs.pretty_print_envs(envs.get_global_envs()))
 
         train_mode = envs.get_global_env("train.trainer")
+
         if train_mode == "SingleTraining":
             trainer = SingleTrainer(yaml_path)
         elif train_mode == "ClusterTraining":
             trainer = ClusterTrainer(yaml_path)
         elif train_mode == "CtrTrainer":
             trainer = CtrPaddleTrainer(config)
+        elif train_mode == "UserDefineTrainer":
+            train_location = envs.get_global_env("train.trainer.location")
+            train_dirname = os.path.dirname(train_location)
+            base_name = os.path.splitext(os.path.basename(train_location))[0]
+            sys.path.append(train_dirname)
+            trainer_class = envs.lazy_instance(base_name, "UserDefineTrainer")
+            trainer = trainer_class(yaml_path)
         else:
             raise ValueError("trainer only support SingleTraining/ClusterTraining")
-
         return trainer
 
     @staticmethod