Fix CIFAR num_samples (#432)

Lightning-Universe · Dec 8, 2020 · 5f748b5 · 5f748b5
1 parent dd31a45
commit 5f748b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):