zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

check_anchors() bug fix #102

This commit is contained in:
Glenn Jocher 2020-06-16 10:34:16 -07:00
parent 7c6fd4d670
commit 05b8ee5ca4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

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