From 8257136012e082ca169ae7aa14e09fd2cbc8091c Mon Sep 17 00:00:00 2001 From: dongdaxiang Date: Thu, 11 Apr 2019 19:20:27 +0800 Subject: [PATCH] fix bug test=develop --- python/paddle/fluid/trainer_desc.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/trainer_desc.py b/python/paddle/fluid/trainer_desc.py index c082a2b4d12..c742ee002aa 100644 --- a/python/paddle/fluid/trainer_desc.py +++ b/python/paddle/fluid/trainer_desc.py @@ -78,7 +78,7 @@ class MultiTrainer(TrainerDesc): def _gen_trainer_desc(self): super(MultiTrainer, self)._gen_trainer_desc() self.proto_desc.class_name = "MultiTrainer" - self._device_worker._set_infer(self.infer_) + self._device_worker._set_infer(self._infer) self._device_worker._gen_worker_desc(self.proto_desc) @@ -96,6 +96,6 @@ class DistMultiTrainer(TrainerDesc): self.proto_desc.class_name = "DistMultiTrainer" if self._program == None: raise RuntimeError("None Program") - self._device_worker._set_infer(self.infer_) - self._device_worker._set_program(self.program_) + self._device_worker._set_infer(self._infer) + self._device_worker._set_program(self._program) self._device_worker._gen_worker_desc(self.proto_desc) -- GitLab