[Fix] Fix resize mix argument bug.

open-mmlab · Jan 4, 2024 · e76681e · e76681e
1 parent 9ac4b31
commit e76681e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmpretrain/models/utils/batch_augments/resizemix.py b/mmpretrain/models/utils/batch_augments/resizemix.py
@@ -87,7 +87,7 @@ def mix(self, batch_inputs: torch.Tensor,
         (y1, y2, x1, x2), lam = self.cutmix_bbox_and_lam(img_shape, lam)
         batch_inputs[:, :, y1:y2, x1:x2] = F.interpolate(
             batch_inputs[index],
-            size=(y2 - y1, x2 - x1),
+            size=(int(y2 - y1), int(x2 - x1)),
             mode=self.interpolation,
             align_corners=False)
         mixed_scores = lam * batch_scores + (1 - lam) * batch_scores[index, :]