Fix bugs that Mixup does not work device=cpu

ground0state · ground0state · commit 047a5ec05f26 · 2021-10-12T23:51:46.000+09:00
diff --git a/timm/data/mixup.py b/timm/data/mixup.py
@@ -214,7 +214,7 @@ def __call__(self, x, target):
             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
 
 
