diff --git a/models.py b/models.py
index b3c9a197..c924aa0b 100755
--- a/models.py
+++ b/models.py
@@ -139,8 +139,8 @@ class YOLOLayer(nn.Module):
 
         # 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: