We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ab9bcd8 commit 092a6a5Copy full SHA for 092a6a5
tools/train.py
@@ -63,11 +63,16 @@ def parse_config():
63
64
cfg = create_cfg_from_sets(args.cfg_file, args.modify_cfgs, args.set_cfgs, cfg)
65
66
- cfg.TAG = Path(args.cfg_file).stem
67
cfg.EXP_GROUP_PATH = '/'.join(args.cfg_file.split('/')[1:-1]) # remove 'cfgs' and 'xxxx.yaml'
68
69
args.use_amp = args.use_amp or cfg.OPTIMIZATION.get('USE_AMP', False)
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
+
76
return args, cfg
77
78
0 commit comments