multi-gpu ckpt filesize bug fix

pull/384/head
Glenn Jocher 2020-07-11 12:39:27 -07:00
parent a586751904
commit 01a73ec08e
1 changed files with 1 additions and 1 deletions

View File

@ -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,