code clean

fleetrec/models/base.py

@@ -99,30 +99,6 @@ class Model(object):
         """
         pass
 
-    @abc.abstractmethod
-    def shrink(self, params):
-        """R
-        """
-        pass
-
-    @abc.abstractmethod
-    def dump_model_program(self, path):
-        """R
-        """
-        pass
-
-    @abc.abstractmethod
-    def dump_inference_param(self, params):
-        """R
-        """
-        pass
-
-    @abc.abstractmethod
-    def dump_inference_program(self, inference_layer, path):
-        """R
-        """
-        pass
-
 
 class YamlModel(Model):
     """R

fleetrec/trainer/transpiler_trainer.py

@@ -115,7 +115,7 @@ class TranspileTrainer(Trainer):
 
     def instance(self, context):
         models = envs.get_global_env("train.model.models")
-        model_class = envs.lazy_instance(models, "TrainNet")
+        model_class = envs.lazy_instance(models, "TrainModel")
         self.model = model_class(None)
         context['status'] = 'init_pass'