Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-27 12:14:01 -08:00
parent 59de209ab2
commit 7ef7501c36
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -436,7 +436,7 @@ def compute_loss(p, targets, model): # predictions, targets, model
lcls *= h['cls'] lcls *= h['cls']
if red == 'sum': if red == 'sum':
lbox *= 3 / ng lbox *= 3 / ng
lobj *= 3 / np lobj *= 3 / np * 2
lcls *= 3 / ng / model.nc lcls *= 3 / ng / model.nc
loss = lbox + lobj + lcls loss = lbox + lobj + lcls