updates

ultralytics · Nov 9, 2018 · e04bb75 · e04bb75
1 parent 966bc16
commit e04bb75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models.py b/models.py
@@ -139,8 +139,8 @@ def forward(self, p, targets=None, requestPrecision=False):
 
         # Training
         if targets is not None:
-            MSELoss = nn.MSELoss(size_average=True)
-            BCEWithLogitsLoss = nn.BCEWithLogitsLoss(size_average=True)
+            MSELoss = nn.MSELoss()
+            BCEWithLogitsLoss = nn.BCEWithLogitsLoss()
             CrossEntropyLoss = nn.CrossEntropyLoss()
 
             if requestPrecision: