updates

2019-07-10 22:11:48 +02:00 · 2019-07-10 22:11:48 +02:00 · 3373006d0e
parent 4f6ef59d92
commit 3373006d0e
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -15,7 +15,7 @@ from utils.utils import *
 hyp = {'giou': 1.666,  # giou loss gain
       'xy': 4.062,  # xy loss gain
       'wh': 0.1845,  # wh loss gain
-       'cls': 42.6,  # cls loss gain
+       'cls': 1.0,  # cls loss gain
       'cls_pw': 3.34,  # cls BCELoss positive_weight
       'obj': 12.61,  # obj loss gain
       'obj_pw': 8.338,  # obj BCELoss positive_weight