diff --git a/fleetrec/models/base.py b/fleetrec/models/base.py index 99e822dcb8f8984a355db999256f9143883f7476..0a92dee06030b7958efc0bc9ae8992f432849282 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 44571962ffeb67aeeef8f73ad70513629c7fd11f..976ce892797478c19074a207a330cfa4aba3ded6 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'