Upload processor

image_processing_swin.py

@@ -30,9 +30,10 @@ class MyCustomSwinProcessor(ViTImageProcessor):
         bottom = pad_h - top
 
         # Pad with white because its the dataset default background color
-        return transforms.functional.pad(image, (left, top, right, bottom), fill=255).totensor()
+        return transforms.functional.pad(image, (left, top, right, bottom), fill=255)
 
     def preprocess(self, images, **kwargs):
         images = [self.resize_and_pad(image, target_size=224) for image in images]
+        images = [transforms.ToTensor()(image) for image in images]
         images = torch.stack(images)
         return super().preprocess(images, **kwargs)