ChristopherSTAN commited on
Commit
a8d17ce
·
unverified ·
1 Parent(s): d18adf8

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -37,7 +37,7 @@ hyp = {'lr0': 0.01, # initial learning rate (SGD=1E-2, Adam=1E-3)
37
  'obj_pw': 1.0, # obj BCELoss positive_weight
38
  'iou_t': 0.20, # iou training threshold
39
  'anchor_t': 4.0, # anchor-multiple threshold
40
- 'fl_gamma': 0.0, # focal loss gamma (efficientDet default is gamma=1.5)
41
  'hsv_h': 0.014, # image HSV-Hue augmentation (fraction)
42
  'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction)
43
  'hsv_v': 0.36, # image HSV-Value augmentation (fraction)
 
37
  'obj_pw': 1.0, # obj BCELoss positive_weight
38
  'iou_t': 0.20, # iou training threshold
39
  'anchor_t': 4.0, # anchor-multiple threshold
40
+ 'fl_gamma': 1.5, # focal loss gamma (efficientDet default is gamma=1.5)
41
  'hsv_h': 0.014, # image HSV-Hue augmentation (fraction)
42
  'hsv_s': 0.68, # image HSV-Saturation augmentation (fraction)
43
  'hsv_v': 0.36, # image HSV-Value augmentation (fraction)