dump config before training (#282)

2021-06-04 16:26:13 +08:00 · 2021-06-04 16:26:13 +08:00 · 5066e32306
parent 9f7168e770
commit 5066e32306
1 changed files with 2 additions and 0 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -95,6 +95,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')