Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
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, | |||
Loading…