From 933e05cb44605f6ead46c644043673c1d4206a48 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 7 Apr 2020 17:35:35 -0700
Subject: [PATCH] augment update

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

diff --git a/models.py b/models.py
index f574cb84..6345dbf8 100755
--- a/models.py
+++ b/models.py
@@ -252,9 +252,9 @@ class Darknet(nn.Module):
             # for i, yi in enumerate(y):  # coco small, medium, large = < 32**2 < 96**2 <
             #     area = yi[..., 2:4].prod(2)[:, :, None]
             #     if i == 1:
-            #         yi = yi * (area < 96. ** 2).float()
+            #         yi *= (area < 96. ** 2).float()
             #     elif i == 2:
-            #         yi = yi * (area > 32. ** 2).float()
+            #         yi *= (area > 32. ** 2).float()
             #     y[i] = yi
 
             y = torch.cat(y, 1)