Glenn Jocher
hace 4 años
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
+ 1
- 1
train.py
Ver fichero
| @@ -199,7 +199,7 @@ def train(hyp): | |||
| tb_writer.add_histogram('classes', c, 0) | |||
| # 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 | |||
| ema = torch_utils.ModelEMA(model) | |||
Cargando…