From 38178d53e5420c0369105727aec2a7db5a4acf86 Mon Sep 17 00:00:00 2001 From: Yang Zhang Date: Sun, 29 Sep 2019 13:09:58 +0800 Subject: [PATCH] Limit config printing to rand 0 when training with multi processes (#3448) --- tools/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/train.py b/tools/train.py index 2745c7f7f..159cacc58 100644 --- a/tools/train.py +++ b/tools/train.py @@ -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() -- GitLab