Skip to content

Commit 092a6a5

Browse files
author
David Josef Emmerichs
committed
make train TAG more descriptive based on modifier and setter
1 parent ab9bcd8 commit 092a6a5

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tools/train.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,11 +63,16 @@ def parse_config():
6363

6464
cfg = create_cfg_from_sets(args.cfg_file, args.modify_cfgs, args.set_cfgs, cfg)
6565

66-
cfg.TAG = Path(args.cfg_file).stem
6766
cfg.EXP_GROUP_PATH = '/'.join(args.cfg_file.split('/')[1:-1]) # remove 'cfgs' and 'xxxx.yaml'
6867

6968
args.use_amp = args.use_amp or cfg.OPTIMIZATION.get('USE_AMP', False)
7069

70+
cfg.TAG = Path(args.cfg_file).stem
71+
if len(args.modify_cfgs):
72+
cfg.TAG += '__' + '__'.join([Path(m).stem for m in args.modify_cfgs])
73+
if len(args.set_cfgs):
74+
cfg.TAG += '__s__' + '__'.join([s.replace('/', '_') for s in args.set_cfgs])
75+
7176
return args, cfg
7277

7378

0 commit comments

Comments
 (0)