From b8a6afa70e47b1f130aa80f3688643d9681cccf9 Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Sun, 29 Aug 2021 03:25:47 -0700 Subject: [PATCH] Fix ImageNet val loader to use val transform (#713) Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> --- CHANGELOG.md | 2 ++ pl_bolts/datamodules/imagenet_datamodule.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) 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,