mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

do not save eval_only to config.yaml

Browse Source
This commit is contained in:
zuchen.wang 2021-11-24 16:01:04 +08:00
parent 3cad97418b
commit d1f8dbb308
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
projects/Shoe/train_net.py
View File

@ -24,14 +24,15 @@ def setup(args):
cfg = get_cfg() cfg = get_cfg()
cfg.merge_from_file(args.config_file) cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts) cfg.merge_from_list(args.opts)
setattr(cfg, 'eval_only', args.eval_only)
cfg.freeze()
default_setup(cfg, args) default_setup(cfg, args)
return cfg return cfg
def main(args): def main(args):
cfg = setup(args) cfg = setup(args)
cfg.defrost()
setattr(cfg, 'eval_only', args.eval_only)
cfg.freeze()
if args.eval_only: if args.eval_only:
cfg.defrost() cfg.defrost()