diff --git a/tools/train.py b/tools/train.py index 1321d0578..1e1d01ac1 100644 --- a/tools/train.py +++ b/tools/train.py @@ -109,9 +109,6 @@ def main(): if args.cfg_options is not None: cfg.merge_from_dict(args.cfg_options) - # set multi-process settings - setup_multi_processes(cfg) - # set cudnn_benchmark if cfg.get('cudnn_benchmark', False): torch.backends.cudnn.benchmark = True @@ -163,6 +160,9 @@ def main(): log_file = osp.join(cfg.work_dir, f'{timestamp}.log') logger = get_root_logger(log_file=log_file, log_level=cfg.log_level) + # set multi-process settings + setup_multi_processes(cfg) + # init the meta dict to record some important information such as # environment info and seed, which will be logged meta = dict()