From 5a09a5bf421cd45d1c834176fa6184a5bafa4143 Mon Sep 17 00:00:00 2001
From: Kai Zhao <694691@qq.com>
Date: Mon, 9 Jan 2023 21:53:05 -0800
Subject: [PATCH] dump config (#1584)

---
 tools/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/train.py b/tools/train.py
index 3bb7b5d5b6..b35db8583a 100644
--- 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')