diff --git a/train.py b/train.py index a38b9775..ae1e0405 100644 --- a/train.py +++ b/train.py @@ -20,7 +20,7 @@ last = wdir + 'last.pt' best = wdir + 'best.pt' results_file = 'results.txt' -# Hyperparameters (k-series, 53.3 mAP yolov3-spp-320) https://github.com/ultralytics/yolov3/issues/310 +# Hyperparameters (k-series, 57.7 mAP yolov3-spp-416) https://github.com/ultralytics/yolov3/issues/310 hyp = {'giou': 3.31, # giou loss gain 'cls': 42.4, # cls loss gain 'cls_pw': 1.0, # cls BCELoss positive_weight