未验证 提交 9914dff7 编写于 作者: W WangXi 提交者: GitHub

[hybrid performance] optimize pipeline performance

上级 43876e8b
...@@ -1506,7 +1506,8 @@ class Executor(object): ...@@ -1506,7 +1506,8 @@ class Executor(object):
trainer._set_infer(is_infer) trainer._set_infer(is_infer)
trainer._gen_trainer_desc() trainer._gen_trainer_desc()
self._dump_debug_info(program=program, trainer=trainer) if program._pipeline_opt is None:
self._dump_debug_info(program=program, trainer=trainer)
# in case of calling _set_use_ps_gpu explicitly # in case of calling _set_use_ps_gpu explicitly
if dataset.use_ps_gpu is False: if dataset.use_ps_gpu is False:
dataset._set_use_ps_gpu(trainer.proto_desc.use_ps_gpu) dataset._set_use_ps_gpu(trainer.proto_desc.use_ps_gpu)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册