Anchor override (#2350)
This commit is contained in:
1 parent
fe6ebb96bb
commit
a3ecf0fd64
2 files changed
+6
-3
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user