From adf823851e7c2336cdab06726d11bb5c65228023 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Thu, 20 Jul 2023 23:25:26 +0200
Subject: [PATCH] [models.pasa] Set grayscale/resize model transforms
---
src/ptbench/models/pasa.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/ptbench/models/pasa.py b/src/ptbench/models/pasa.py
index e2cb9b05..650aa7d4 100644
--- a/src/ptbench/models/pasa.py
+++ b/src/ptbench/models/pasa.py
@@ -73,7 +73,8 @@ class Pasa(pl.LightningModule):
self.name = "pasa"
self.model_transforms = [
- torchvision.transforms.Resize(512),
+ torchvision.transforms.Grayscale(),
+ torchvision.transforms.Resize(512, antialias=True),
]
self._train_loss = train_loss
--
GitLab