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

train.py iou_t to 0.20

This commit is contained in:
Glenn Jocher 2020-04-22 11:34:34 -07:00
parent 77b3829d56
commit 7cbac5a3ea
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

@ -28,7 +28,7 @@ hyp = {'giou': 3.54, # giou loss gain
'cls_pw': 1.0, # cls BCELoss positive_weight 'cls_pw': 1.0, # cls BCELoss positive_weight
'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320) 'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320)
'obj_pw': 1.0, # obj BCELoss positive_weight 'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.1, # iou training threshold 'iou_t': 0.20, # iou training threshold
'lr0': 0.01, # initial learning rate (SGD=5E-3, Adam=5E-4) 'lr0': 0.01, # initial learning rate (SGD=5E-3, Adam=5E-4)
'lrf': 0.0005, # final learning rate (with cos scheduler) 'lrf': 0.0005, # final learning rate (with cos scheduler)
'momentum': 0.937, # SGD momentum 'momentum': 0.937, # SGD momentum