BUG Fix: ensure the right loss function is selected when mixtup or cutmix is enabeled

2021-05-02 10:27:25 -04:00 · 2021-05-02 10:27:25 -04:00 · 2f76ec64fd
parent 686507a6a5
commit 2f76ec64fd
1 changed files with 1 additions and 1 deletions
--- 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: