Skip to content
Snippets Groups Projects
Commit a4686009 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[models.pasa] Set grayscale/resize model transforms

parent 249274a9
No related branches found
No related tags found
1 merge request!9Update Montgomery dataset
...@@ -73,7 +73,8 @@ class Pasa(pl.LightningModule): ...@@ -73,7 +73,8 @@ class Pasa(pl.LightningModule):
self.name = "pasa" self.name = "pasa"
self.model_transforms = [ self.model_transforms = [
torchvision.transforms.Resize(512), torchvision.transforms.Grayscale(),
torchvision.transforms.Resize(512, antialias=True),
] ]
self._train_loss = train_loss self._train_loss = train_loss
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment