Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
View File
| tb_writer.add_histogram('classes', c, 0) | tb_writer.add_histogram('classes', c, 0) | ||||
| # Check anchors | # Check anchors | ||||
| check_anchors(dataset, model=model.model[-1].anchor_grid, thr=hyp['anchor_t'], imgsz=imgsz) | |||||
| check_anchors(dataset, model=model, thr=hyp['anchor_t'], imgsz=imgsz) | |||||
| # Exponential moving average | # Exponential moving average | ||||
| ema = torch_utils.ModelEMA(model) | ema = torch_utils.ModelEMA(model) |
Loading…