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): ...@@ -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.
Finish editing this message first!
Please register or to comment