Anchor override (#2350)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-03-02 23:08:21 -08:00
1 parent fe6ebb96bb
commit a3ecf0fd64
2 files changed
+6 -3

No files matched your search

+1 -1
View File
@@ -84,7 +84,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
model.load_state_dict(state_dict, strict=False) # load
logger.info('Transferred %g/%g items from %s' % (len(state_dict), len(model.state_dict()), weights)) # report
else:
model = Model(opt.cfg, ch=3, nc=nc).to(device) # create
model = Model(opt.cfg, ch=3, nc=nc, anchors=hyp.get('anchors')).to(device) # create
# Freeze
freeze = [] # parameter names to freeze (full or partial)