add config dump in training (open-mmlab#2779)

Pooja0509 · May 20, 2020 · 227d338 · 227d338
1 parent 3b908f5
commit 227d338
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/train.py b/tools/train.py
@@ -100,6 +100,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')