未验证 提交 3250437b 编写于 作者: Y Yang Zhang 提交者: GitHub

Limit config printing to rand 0 when training with multi processes (#3448)

上级 efc994ec
......@@ -81,7 +81,8 @@ def main():
# check if set use_gpu=True in paddlepaddle cpu version
check_gpu(cfg.use_gpu)
print_total_cfg(cfg)
if not FLAGS.dist or trainer_id == 0:
print_total_cfg(cfg)
if cfg.use_gpu:
devices_num = fluid.core.get_cuda_device_count()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册