updates

2019-08-27 12:57:19 +02:00 · 2019-08-27 12:57:19 +02:00 · c906047db3
parent 798a7396f1
commit c906047db3
1 changed files with 2 additions and 2 deletions
--- 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()