Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Jun 16, 2020

Commit

5a50491

•

1 Parent(s): 05b8ee5

check_anchors bug fix

Files changed (1) hide show

train.py CHANGED Viewed

@@ -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)
     # Exponential moving average
     ema = torch_utils.ModelEMA(model)

     tb_writer.add_histogram('classes', c, 0)
     # Check anchors
+    check_anchors(dataset, model=model, thr=hyp['anchor_t'], imgsz=imgsz)
     # Exponential moving average
     ema = torch_utils.ModelEMA(model)