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

[models.pasa] Set grayscale/resize model transforms

parent 10cd1e24
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
......@@ -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