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

updates

This commit is contained in:
Glenn Jocher 2019-08-27 12:57:19 +02:00
parent 798a7396f1
commit c906047db3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -45,8 +45,8 @@ def train():
weights = opt.weights # initial training weights weights = opt.weights # initial training weights
if 'pw' not in opt.arc: # remove BCELoss positive weights if 'pw' not in opt.arc: # remove BCELoss positive weights
hyp['cls_pw'] = 0 hyp['cls_pw'] = 1.
hyp['obj_pw'] = 0 hyp['obj_pw'] = 1.
# Initialize # Initialize
init_seeds() init_seeds()