From cca620208eff32f40c857a2b6ea5d6bd7919264e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 16 Feb 2020 23:13:34 -0800 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index cb17b946..917fbda8 100644 --- a/train.py +++ b/train.py @@ -25,7 +25,7 @@ results_file = 'results.txt' hyp = {'giou': 3.54, # giou loss gain 'cls': 37.4, # cls loss gain 'cls_pw': 1.0, # cls BCELoss positive_weight - 'obj': 49.5, # obj loss gain (*=img_size/320 if img_size != 320) + 'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320) 'obj_pw': 1.0, # obj BCELoss positive_weight 'iou_t': 0.225, # iou training threshold 'lr0': 0.00579, # initial learning rate (SGD=5E-3, Adam=5E-4)