Glenn Jocher
преди 4 години
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
+ 1
- 1
train.py
Целия файл
| @@ -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) | |||
Loading…