Cifar also returns real data length (#211)

Author: keon

torchvision/datasets/cifar.py

@@ -128,9 +128,9 @@ def __getitem__(self, index):
 
     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