From 3373006d0ecfedf4fae13d7ccc6cc5131645f294 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 10 Jul 2019 22:11:48 +0200
Subject: [PATCH] updates

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 8b297636..4c4dda78 100644
--- 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