check_anchors() bug fix #90

This commit is contained in:
Glenn Jocher committed 2020-06-16 10:14:04 -07:00
1 parent e1e3399286
commit ec81c7b5f2
2 files changed
+7 -7

No files matched your search

+1 -1
View File
@@ -199,7 +199,7 @@ def train(hyp):
tb_writer.add_histogram('classes', c, 0)
# Check anchors
check_best_possible_recall(dataset, anchors=model.model[-1].anchor_grid, thr=hyp['anchor_t'], imgsz=imgsz)
check_anchors(dataset, anchors=model.model[-1].anchor_grid, thr=hyp['anchor_t'], imgsz=imgsz)
# Exponential moving average
ema = torch_utils.ModelEMA(model)