From fd71fe8451986ee51808f0809a4a833896c4b4c5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 22 Aug 2020 16:18:53 -0700 Subject: [PATCH] Update hyp.finetune.yaml --- data/hyp.finetune.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/hyp.finetune.yaml b/data/hyp.finetune.yaml index 3c90a26..ac350f6 100644 --- a/data/hyp.finetune.yaml +++ b/data/hyp.finetune.yaml @@ -7,9 +7,9 @@ lr0: 0.01 # initial learning rate (SGD=1E-2, Adam=1E-3) momentum: 0.94 # SGD momentum/Adam beta1 weight_decay: 0.0005 # optimizer weight decay 5e-4 giou: 0.05 # GIoU loss gain -cls: 0.2 # cls loss gain +cls: 0.4 # cls loss gain cls_pw: 1.0 # cls BCELoss positive_weight -obj: 0.3 # obj loss gain (scale with pixels) +obj: 0.5 # obj loss gain (scale with pixels) obj_pw: 1.0 # obj BCELoss positive_weight iou_t: 0.20 # IoU training threshold anchor_t: 4.0 # anchor-multiple threshold