diff --git a/tools/train.py b/tools/train.py index 64e29ec3..d461c4b1 100644 --- 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')