Skip to content

Commit

Permalink
Hardcoding shuffle=False for val and test
Browse files Browse the repository at this point in the history
  • Loading branch information
briankosw committed Dec 15, 2020
1 parent dd234e1 commit ebaaf18
Show file tree
Hide file tree
Showing 11 changed files with 23 additions and 23 deletions.
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/binary_mnist_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset_val,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -161,7 +161,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/cifar10_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset_val,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
pin_memory=self.pin_memory,
drop_last=self.drop_last
Expand All @@ -179,7 +179,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/cityscapes_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
pin_memory=self.pin_memory,
drop_last=self.drop_last
Expand All @@ -185,7 +185,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/fashion_mnist_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset_val,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -158,7 +158,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/imagenet_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -208,7 +208,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/kitti_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def val_dataloader(self):
loader = DataLoader(
self.valset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -127,7 +127,7 @@ def test_dataloader(self):
loader = DataLoader(
self.testset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory,
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/mnist_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset_val,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory,
Expand All @@ -155,7 +155,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory,
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/sklearn_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,7 +215,7 @@ def val_dataloader(self):
loader = DataLoader(
self.val_dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -226,7 +226,7 @@ def test_dataloader(self):
loader = DataLoader(
self.test_dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
4 changes: 2 additions & 2 deletions pl_bolts/datamodules/ssl_imagenet_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def val_dataloader(self, num_images_per_class=50, add_normalize=False):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -129,7 +129,7 @@ def test_dataloader(self, num_images_per_class, add_normalize=False):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
8 changes: 4 additions & 4 deletions pl_bolts/datamodules/stl10_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def val_dataloader(self):
loader = DataLoader(
dataset_val,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drpo_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down Expand Up @@ -234,7 +234,7 @@ def val_dataloader_mixed(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand All @@ -255,7 +255,7 @@ def test_dataloader(self):
loader = DataLoader(
dataset,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down Expand Up @@ -294,7 +294,7 @@ def val_dataloader_labeled(self):
loader = DataLoader(
labeled_val,
batch_size=self.batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory
Expand Down
2 changes: 1 addition & 1 deletion pl_bolts/datamodules/vocdetection_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ def val_dataloader(self, batch_size=1, transforms=None):
loader = DataLoader(
dataset,
batch_size=batch_size,
shuffle=self.shuffle,
shuffle=False,
num_workers=self.num_workers,
drop_last=self.drop_last,
pin_memory=self.pin_memory,
Expand Down

0 comments on commit ebaaf18

Please sign in to comment.