diff --git a/tools/train.py b/tools/train.py index 2745c7f7f2bf7cd924050c03effe687d77e8ea28..159cacc586f4a1fd20bd333c25603b6138a347fd 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()