mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

multi-gpu ckpt filesize bug fix #253

Browse Source
This commit is contained in:
Glenn Jocher 2020-07-11 12:39:27 -07:00
parent a586751904
commit 01a73ec08e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
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,