From 33edc01ab1c7b3a74befb2e4428bf4f781a076d0 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Tue, 14 Apr 2020 11:46:28 +0800
Subject: [PATCH] code clean

---
 fleetrec/models/base.py                | 24 ------------------------
 fleetrec/trainer/transpiler_trainer.py |  2 +-
 2 files changed, 1 insertion(+), 25 deletions(-)

diff --git a/fleetrec/models/base.py b/fleetrec/models/base.py
index 99e822dc..0a92dee0 100644
--- a/fleetrec/models/base.py
+++ b/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
diff --git a/fleetrec/trainer/transpiler_trainer.py b/fleetrec/trainer/transpiler_trainer.py
index 44571962..976ce892 100644
--- a/fleetrec/trainer/transpiler_trainer.py
+++ b/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'
 
-- 
GitLab