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

fix platform

上级 2831d752
......@@ -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 = []
......
......@@ -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"
......
......@@ -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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册