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

updates

This commit is contained in:
Glenn Jocher 2020-01-19 14:59:07 -08:00
parent 723431d6c3
commit 51b5b3288f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -20,7 +20,7 @@ last = wdir + 'last.pt'
best = wdir + 'best.pt' best = wdir + 'best.pt'
results_file = 'results.txt' results_file = 'results.txt'
# Hyperparameters (results68: 59.2 mAP@0.5 yolov3-spp-416) https://github.com/ultralytics/yolov3/issues/310 # Hyperparameters (results68: 59.9 mAP@0.5 yolov3-spp-416) https://github.com/ultralytics/yolov3/issues/310
hyp = {'giou': 3.54, # giou loss gain hyp = {'giou': 3.54, # giou loss gain
'cls': 37.4, # cls loss gain 'cls': 37.4, # cls loss gain
@ -28,7 +28,7 @@ hyp = {'giou': 3.54, # giou loss gain
'obj': 49.5, # obj loss gain (*=img_size/320 if img_size != 320) 'obj': 49.5, # 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.225, # iou training threshold 'iou_t': 0.225, # iou training threshold
'lr0': 0.00579, # initial learning rate (SGD=1E-3, Adam=9E-5) 'lr0': 0.00579, # initial learning rate (SGD=5E-3, Adam=5E-4)
'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf) 'lrf': -4., # final LambdaLR learning rate = lr0 * (10 ** lrf)
'momentum': 0.937, # SGD momentum 'momentum': 0.937, # SGD momentum
'weight_decay': 0.000484, # optimizer weight decay 'weight_decay': 0.000484, # optimizer weight decay
@ -94,6 +94,7 @@ def train():
pg0 += [v] # all else pg0 += [v] # all else
if opt.adam: if opt.adam:
hyp['lr0'] *= 0.1 # reduce lr (i.e. SGD=5E-3, Adam=5E-4)
optimizer = optim.Adam(pg0, lr=hyp['lr0']) optimizer = optim.Adam(pg0, lr=hyp['lr0'])
# optimizer = AdaBound(pg0, lr=hyp['lr0'], final_lr=0.1) # optimizer = AdaBound(pg0, lr=hyp['lr0'], final_lr=0.1)
else: else: