Skip to content

Commit ab9bcd8

Browse files
author
David Josef Emmerichs
committed
save not only base configuration file but also modified configuration
1 parent a23d733 commit ab9bcd8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tools/train.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,9 @@ def main():
115115
logger.info('{:16} {}'.format(key, val))
116116
log_config_to_file(cfg, logger=logger)
117117
if cfg.LOCAL_RANK == 0:
118-
os.system('cp %s %s' % (args.cfg_file, output_dir))
118+
os.system('cp %s %s' % (args.cfg_file, output_dir / ("BASE_" + Path(args.cfg_file).name)))
119+
set_cfg = create_cfg_from_sets(args.cfg_file, args.modify_cfgs, args.set_cfgs)
120+
cfg_to_yaml_file(set_cfg, output_dir / Path(args.cfg_file).name)
119121

120122
tb_log = SummaryWriter(log_dir=str(output_dir / 'tensorboard')) if cfg.LOCAL_RANK == 0 else None
121123

0 commit comments

Comments
 (0)