diff --git a/tools/train.py b/tools/train.py index 3bb7b5d5b6..b35db8583a 100644 --- a/tools/train.py +++ b/tools/train.py @@ -104,6 +104,8 @@ def main(): # create work_dir mmcv.mkdir_or_exist(osp.abspath(cfg.work_dir)) + # dump config + cfg.dump(osp.join(cfg.work_dir, osp.basename(args.config))) # init the logger before other steps timestamp = time.strftime('%Y%m%d_%H%M%S', time.localtime()) log_file = osp.join(cfg.work_dir, f'{timestamp}.log')