updates

2018-11-10 00:54:55 +01:00
parent 966bc16d1a
commit e04bb75ff1
1 changed files with 2 additions and 2 deletions
@@ -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: