Merge pull request #913 from ground0state/master

Fix bugs that Mixup does not work when device is cpu
3 years ago · 2c33ca6d8c
parent cd34913278 047a5ec05f
commit 2c33ca6d8c
1 changed files with 1 additions and 1 deletions
--- a/timm/data/mixup.py
+++ b/timm/data/mixup.py
@ -214,7 +214,7 @@ class Mixup:
            lam = self._mix_pair(x)
        else:
            lam = self._mix_batch(x)
-        target = mixup_target(target, self.num_classes, lam, self.label_smoothing)
+        target = mixup_target(target, self.num_classes, lam, self.label_smoothing, x.device)
        return x, target