multi-gpu ckpt filesize bug fix #253
parent
a586751904
commit
01a73ec08e
2
train.py
2
train.py
|
@ -287,7 +287,7 @@ def train(hyp):
|
|||
scheduler.step()
|
||||
|
||||
# mAP
|
||||
ema.update_attr(model, include=['md', 'nc', 'hyp', 'names', 'stride'])
|
||||
ema.update_attr(model, include=['md', 'nc', 'hyp', 'gr', 'names', 'stride'])
|
||||
final_epoch = epoch + 1 == epochs
|
||||
if not opt.notest or final_epoch: # Calculate mAP
|
||||
results, maps, times = test.test(opt.data,
|
||||
|
|
Loading…
Reference in New Issue