diff --git a/pl_bolts/datamodules/cifar10_datamodule.py b/pl_bolts/datamodules/cifar10_datamodule.py index e316e5fe14..d8822591dd 100644 --- a/pl_bolts/datamodules/cifar10_datamodule.py +++ b/pl_bolts/datamodules/cifar10_datamodule.py @@ -95,7 +95,7 @@ def __init__( self.batch_size = batch_size self.seed = seed self.data_dir = data_dir if data_dir is not None else os.getcwd() - self.num_samples = 60000 - val_split + self.num_samples = 50000 - val_split @property def num_classes(self):