fix bug on device

pipeline.py

@@ -6059,8 +6059,18 @@ class MaxLongEdgeMinShortEdgeResize(torch.nn.Module):
         if max(new_width, new_height) > self.max_size:
             scale = self.max_size / max(new_width, new_height)
             new_width, new_height = self._apply_scale(new_width, new_height, scale)
-
-        return F.resize(img, (new_height, new_width), self.interpolation, antialias=self.antialias)
+        
+        if img.dim() == 3:
+            img = img.unsqueeze(0)  # → [1,C,H,W]
+        resized = F.interpolate(
+            img,
+            size=(new_h, new_w),
+            mode=interpolation,     # e.g. "bilinear"
+            antialias=True,         # if you need anti‑aliasing
+        )
+        resized = resized.squeeze(0)  # → [C,H,W]
+        return resized
+        # return F.resize(img, (new_height, new_width), self.interpolation, antialias=self.antialias)
 
 
 class ImageTransform: