From 2f76ec64fdd1626f80e31c7cfae71521aebb2fa6 Mon Sep 17 00:00:00 2001
From: Maxwell Aladago <maxwell.aladago@gmail.com>
Date: Sun, 2 May 2021 10:27:25 -0400
Subject: [PATCH] BUG Fix: ensure the right loss function is selected when
 mixtup or cutmix is enabeled

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 5db8dd3..55754fe 100644
--- a/main.py
+++ b/main.py
@@ -310,7 +310,7 @@ def main(args):
 
     criterion = LabelSmoothingCrossEntropy()
 
-    if mixup_active > 0.:
+    if mixup_active:
         # smoothing is handled with mixup label transform
         criterion = SoftTargetCrossEntropy()
     elif args.smoothing: