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):
self.name = "pasa"
self.model_transforms = [
torchvision.transforms.Resize(512),
torchvision.transforms.Grayscale(),
torchvision.transforms.Resize(512, antialias=True),
]
self._train_loss = train_loss
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment