diff --git a/fleetrec/examples/ctr-dnn_train_cluster.yaml b/fleetrec/examples/ctr-dnn_train_cluster.yaml index 8eb9eb1e0e53e982cd831c389fdef6cc3f83ab63..53f032534cae69707a850e7142d29828f5d3fc6c 100644 --- a/fleetrec/examples/ctr-dnn_train_cluster.yaml +++ b/fleetrec/examples/ctr-dnn_train_cluster.yaml @@ -41,8 +41,8 @@ train: reader: mode: "dataset" batch_size: 2 - pipe_command: "python /paddle/fleetrec/models/ctr_dnn/dataset.py" - train_data_path: "/paddle/fleetrec/models/ctr_dnn/data/train" + pipe_command: "python /paddle/eleps/fleetrec/models/ctr_dnn/dataset.py" + train_data_path: "/paddle/eleps/fleetrec/models/ctr_dnn/data/train" model: models: "fleetrec.models.ctr_dnn.model" diff --git a/fleetrec/examples/ctr-dnn_train_single.yaml b/fleetrec/examples/ctr-dnn_train_single.yaml index 4091c78ecadd0f9346f1ce8c1eeb835c9ed216ea..985adebe9ea8872dc556ccaf48775f75716b318d 100644 --- a/fleetrec/examples/ctr-dnn_train_single.yaml +++ b/fleetrec/examples/ctr-dnn_train_single.yaml @@ -35,8 +35,8 @@ train: reader: mode: "dataset" batch_size: 2 - pipe_command: "python /paddle/fleetrec/models/ctr_dnn/dataset.py" - train_data_path: "/paddle/fleetrec/models/ctr_dnn/data/train" + pipe_command: "python /paddle/eleps/fleetrec/models/ctr_dnn/dataset.py" + train_data_path: "/paddle/eleps/fleetrec/models/ctr_dnn/data/train" model: models: "fleetrec.models.ctr_dnn.model" diff --git a/fleetrec/models/ctr_dnn/model.py b/fleetrec/models/ctr_dnn/model.py index c6b3997102ff42b06b14d8130d4dad2ce5d754b7..f65a60e8203c2b1db52cab6da6f39977d01c11eb 100644 --- a/fleetrec/models/ctr_dnn/model.py +++ b/fleetrec/models/ctr_dnn/model.py @@ -21,7 +21,7 @@ from fleetrec.models.base import Model class Train(Model): def __init__(self, config): - super().__init__(config) + Model.__init__(self, config) self.namespace = "train.model" def input(self): diff --git a/fleetrec/trainer/transpiler_trainer.py b/fleetrec/trainer/transpiler_trainer.py index 2c6e6802efd57e167756f2579694d8e249fccfcb..3d1fa934300023285d0fe2e725245096d616f7e9 100644 --- a/fleetrec/trainer/transpiler_trainer.py +++ b/fleetrec/trainer/transpiler_trainer.py @@ -29,7 +29,7 @@ class TranspileTrainer(Trainer): def __init__(self, config=None): Trainer.__init__(self, config) self.processor_register() - + self.model = None self.inference_models = [] self.increment_models = [] @@ -115,7 +115,7 @@ class TranspileTrainer(Trainer): models = envs.get_global_env("train.model.models") model_package = __import__(models, globals(), locals(), models.split(".")) train_model = getattr(model_package, 'Train') - self.model = train_model() + self.model = train_model(None) context['status'] = 'init_pass' def init(self, context):