提交 431e34de 编写于 作者: T tangwei

fix platform

上级 2831d752
...@@ -64,7 +64,7 @@ class SingleTrainer(TranspileTrainer): ...@@ -64,7 +64,7 @@ class SingleTrainer(TranspileTrainer):
program = fluid.compiler.CompiledProgram( program = fluid.compiler.CompiledProgram(
fluid.default_main_program()).with_data_parallel( 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_varnames = []
metrics_format = [] metrics_format = []
......
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
train: train:
trainer: trainer:
trainer: "fleetrec/demo/user_define_trainer.py" #trainer: "fleetrec/demo/user_define_trainer.py"
threads: 4 threads: 4
# for cluster training # for cluster training
strategy: "async" strategy: "async"
......
...@@ -57,9 +57,9 @@ class Model(ModelBase): ...@@ -57,9 +57,9 @@ class Model(ModelBase):
self._data_var.append(input) self._data_var.append(input)
self._data_var.append(self.label_input) self._data_var.append(self.label_input)
if self._platform != "LINUX": 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) feed_list=self._data_var, capacity=64, use_double_buffer=False, iterable=False)
def net(self): def net(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册