Cifar also returns real data length (#211)

85bb16a8 · Keon · Soumith Chintala · 4aff146c · 85bb16a8
Commit 85bb16a8 authored Jul 27, 2017 by Keon Committed by Soumith Chintala Jul 27, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/datasets/cifar.py torchvision/datasets/cifar.py +2 -2

No files found.
--- a/torchvision/datasets/cifar.py
+++ b/torchvision/datasets/cifar.py
@@ -128,9 +128,9 @@ class CIFAR10(data.Dataset):
    def __len__(self):
        if self.train:
-            return 50000
+            return len(self.train_data)
        else:
-            return 10000
+            return len(self.test_data)
    def _check_integrity(self):
        root = self.root