Added resize transforms in models

src/ptbench/models/alexnet.py

@@ -74,6 +74,7 @@ class Alexnet(pl.LightningModule):
         self.name = "alexnet"
 
         self.model_transforms = [
+            torchvision.transforms.Resize(512),
             torchvision.transforms.ToPILImage(),
             torchvision.transforms.Lambda(lambda x: x.convert("RGB")),
             torchvision.transforms.ToTensor(),

src/ptbench/models/densenet.py

@@ -72,6 +72,7 @@ class Densenet(pl.LightningModule):
         self.name = "densenet-121"
 
         self.model_transforms = [
+            torchvision.transforms.Resize(512),
             torchvision.transforms.ToPILImage(),
             torchvision.transforms.Lambda(lambda x: x.convert("RGB")),
             torchvision.transforms.ToTensor(),

src/ptbench/models/pasa.py

@@ -72,7 +72,9 @@ class Pasa(pl.LightningModule):
 
         self.name = "pasa"
 
-        self.model_transforms = []
+        self.model_transforms = [
+            torchvision.transforms.Resize(512),
+        ]
 
         self._train_loss = train_loss
         self._validation_loss = (