check_anchors bug fix

2020-06-16 10:36:35 -07:00 · 2020-06-16 10:36:35 -07:00 · 5a50491fa1
parent 05b8ee5ca4
commit 5a50491fa1
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/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)