From c906047db3a606441cffa0b8844bb4ae457bf06e Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 27 Aug 2019 12:57:19 +0200
Subject: [PATCH] updates

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 0a56a8c7..01a0eeee 100644
--- a/train.py
+++ b/train.py
@@ -45,8 +45,8 @@ def train():
     weights = opt.weights  # initial training weights
 
     if 'pw' not in opt.arc:  # remove BCELoss positive weights
-        hyp['cls_pw'] = 0
-        hyp['obj_pw'] = 0
+        hyp['cls_pw'] = 1.
+        hyp['obj_pw'] = 1.
 
     # Initialize
     init_seeds()