We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f599d38 commit af88b86Copy full SHA for af88b86
tools/train.py
@@ -109,9 +109,6 @@ def main():
109
if args.cfg_options is not None:
110
cfg.merge_from_dict(args.cfg_options)
111
112
- # set multi-process settings
113
- setup_multi_processes(cfg)
114
-
115
# set cudnn_benchmark
116
if cfg.get('cudnn_benchmark', False):
117
torch.backends.cudnn.benchmark = True
@@ -163,6 +160,9 @@ def main():
163
160
log_file = osp.join(cfg.work_dir, f'{timestamp}.log')
164
161
logger = get_root_logger(log_file=log_file, log_level=cfg.log_level)
165
162
+ # set multi-process settings
+ setup_multi_processes(cfg)
+
166
# init the meta dict to record some important information such as
167
# environment info and seed, which will be logged
168
meta = dict()
0 commit comments