diff --git a/tools/train.py b/tools/train.py index 838c37d9..bfcf6a75 100644 --- a/tools/train.py +++ b/tools/train.py @@ -143,6 +143,9 @@ 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'train_{timestamp}.log')