dump config (#1584)

open-mmlab · Jan 10, 2023 · 5a09a5b · 5a09a5b
1 parent 6ac402e
commit 5a09a5b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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')