From 5a50491fa12a515f829407b2cfc4ec0acb05311d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Jun 2020 10:36:35 -0700 Subject: [PATCH] check_anchors bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index db135a7..484190a 100644 --- 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)