From b3b4ff4107f59a51e7088b587467bba63354649d Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 25 Oct 2019 11:03:33 -0500
Subject: [PATCH] updates
---
train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
index 6cc0ec77..4a43eec6 100644
--- a/train.py
+++ b/train.py
@@ -24,7 +24,7 @@ results_file = 'results.txt'
hyp = {'giou': 3.31, # giou loss gain
'cls': 42.4, # cls loss gain (CE=~1.0, uCE=~20)
'cls_pw': 1.0, # cls BCELoss positive_weight
- 'obj': 50.0, # obj loss gain (*=80 for uBCE with 80 classes)
+ 'obj': 40.0, # obj loss gain (*=80 for uBCE with 80 classes)
'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.213, # iou training threshold
'lr0': 0.00261, # initial learning rate (SGD=1E-3, Adam=9E-5)