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()