From e826153b2d87278263359a78bc41e18253917dfc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?L=C5=91rincz-Moln=C3=A1r=20Szabolcs-Botond?= Date: Tue, 9 Aug 2022 10:50:16 +0300 Subject: [PATCH] Dump training config similar to mmdet. (#410) --- tools/train.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/train.py b/tools/train.py index 838c37d9..bfcf6a75 100644 --- 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')