Dump training config similar to mmdet. (#410)

2022-08-09 10:50:16 +03:00 · 2022-08-09 10:50:16 +03:00 · e826153b2d
parent c2f4be1c8a
commit e826153b2d
1 changed files with 3 additions and 0 deletions
--- 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')