From 431e34de605d112df2cfb937c5875ca483770a81 Mon Sep 17 00:00:00 2001 From: tangwei Date: Wed, 29 Apr 2020 12:56:26 +0800 Subject: [PATCH] fix platform --- fleetrec/core/trainers/single_trainer.py | 2 +- fleetrec/demo/ctr-dnn_train.yaml | 2 +- fleetrec/models/ctr/dnn/model.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fleetrec/core/trainers/single_trainer.py b/fleetrec/core/trainers/single_trainer.py index 56daafe1..c3da7edd 100644 --- a/fleetrec/core/trainers/single_trainer.py +++ b/fleetrec/core/trainers/single_trainer.py @@ -64,7 +64,7 @@ class SingleTrainer(TranspileTrainer): program = fluid.compiler.CompiledProgram( fluid.default_main_program()).with_data_parallel( - loss_name=self.model.get_cost_op.name) + loss_name=self.model.get_cost_op().name) metrics_varnames = [] metrics_format = [] diff --git a/fleetrec/demo/ctr-dnn_train.yaml b/fleetrec/demo/ctr-dnn_train.yaml index 37e1ab22..4d2644e3 100644 --- a/fleetrec/demo/ctr-dnn_train.yaml +++ b/fleetrec/demo/ctr-dnn_train.yaml @@ -14,7 +14,7 @@ train: trainer: - trainer: "fleetrec/demo/user_define_trainer.py" + #trainer: "fleetrec/demo/user_define_trainer.py" threads: 4 # for cluster training strategy: "async" diff --git a/fleetrec/models/ctr/dnn/model.py b/fleetrec/models/ctr/dnn/model.py index ac3887d7..a4e6cf68 100644 --- a/fleetrec/models/ctr/dnn/model.py +++ b/fleetrec/models/ctr/dnn/model.py @@ -57,9 +57,9 @@ class Model(ModelBase): self._data_var.append(input) self._data_var.append(self.label_input) - + if self._platform != "LINUX": - self._data_loader = fluid.io.PyReader( + self._data_loader = fluid.io.DataLoader.from_generator( feed_list=self._data_var, capacity=64, use_double_buffer=False, iterable=False) def net(self): -- GitLab