Fix bugs that Mixup does not work device=cpu

pull/913/head
masafumi 3 years ago
parent cd34913278
commit 047a5ec05f

@ -214,7 +214,7 @@ class Mixup:
lam = self._mix_pair(x) lam = self._mix_pair(x)
else: else:
lam = self._mix_batch(x) 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 return x, target

Loading…
Cancel
Save