Skip to content

P
PaddleDetection

s920243400 / PaddleDetection
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleDetection

通知 2
Star 0
Fork 0
P
PaddleDetection
提交 82571360 编写于 作者: D dongdaxiang
浏览文件
操作

fix bug 

test=develop
上级 a659b37a
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
python/paddle/fluid/trainer_desc.py
浏览文件 @ 82571360
...@@ -78,7 +78,7 @@ class MultiTrainer(TrainerDesc): ...@@ -78,7 +78,7 @@ class MultiTrainer(TrainerDesc):
def _gen_trainer_desc(self): def _gen_trainer_desc(self):
super(MultiTrainer, self)._gen_trainer_desc() super(MultiTrainer, self)._gen_trainer_desc()
self.proto_desc.class_name = "MultiTrainer" 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) self._device_worker._gen_worker_desc(self.proto_desc)
...@@ -96,6 +96,6 @@ class DistMultiTrainer(TrainerDesc): ...@@ -96,6 +96,6 @@ class DistMultiTrainer(TrainerDesc):
self.proto_desc.class_name = "DistMultiTrainer" self.proto_desc.class_name = "DistMultiTrainer"
if self._program == None: if self._program == None:
raise RuntimeError("None Program") raise RuntimeError("None Program")
self._device_worker._set_infer(self.infer_) self._device_worker._set_infer(self._infer)
self._device_worker._set_program(self.program_) self._device_worker._set_program(self._program)
self._device_worker._gen_worker_desc(self.proto_desc) self._device_worker._gen_worker_desc(self.proto_desc)
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或