From e04bb75ff1a77dc3d62f06d85d0a98cb9eb024a7 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 10 Nov 2018 00:54:55 +0100
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index b3c9a1974e..c924aa0b89 100755
--- 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: