diff --git a/CHANGELOG.md b/CHANGELOG.md index 450e09f228..58569e04fd 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -27,6 +27,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Fixed +- Fixed ImageNet val loader to use val transform instead of train transform. + - Fixed the MNIST download giving HTTP 404 with torchvision>=0.9.1 ([#674](https://github.com/PyTorchLightning/lightning-bolts/pull/674)) diff --git a/pl_bolts/datamodules/imagenet_datamodule.py b/pl_bolts/datamodules/imagenet_datamodule.py index 4d641801ec..022f953cd7 100644 --- a/pl_bolts/datamodules/imagenet_datamodule.py +++ b/pl_bolts/datamodules/imagenet_datamodule.py @@ -171,7 +171,7 @@ def val_dataloader(self) -> DataLoader: batch_size: the batch size transforms: the transforms """ - transforms = self.train_transform() if self.val_transforms is None else self.val_transforms + transforms = self.val_transform() if self.val_transforms is None else self.val_transforms dataset = UnlabeledImagenet( self.data_dir,