From 7cbac5a3ea4f6531d471e7a408b340f1a95f844a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 22 Apr 2020 11:34:34 -0700 Subject: [PATCH] train.py iou_t to 0.20 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index b0885308..253f1b18 100644 --- a/train.py +++ b/train.py @@ -28,7 +28,7 @@ hyp = {'giou': 3.54, # giou loss gain 'cls_pw': 1.0, # cls BCELoss positive_weight 'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320) '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) 'lrf': 0.0005, # final learning rate (with cos scheduler) 'momentum': 0.937, # SGD momentum