From 91c7583f64e797466efdfe92bd7cb115e8d4f4ed Mon Sep 17 00:00:00 2001 From: Keon Date: Wed, 26 Jul 2017 11:20:18 -0700 Subject: [PATCH] Return real dataset length --- torchvision/datasets/mnist.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/datasets/mnist.py b/torchvision/datasets/mnist.py index 6f7495141a0..10acef91a5c 100644 --- a/torchvision/datasets/mnist.py +++ b/torchvision/datasets/mnist.py @@ -81,9 +81,9 @@ def __getitem__(self, index): def __len__(self): if self.train: - return 60000 + return len(self.train_data) else: - return 10000 + return len(self.test_data) def _check_exists(self): return os.path.exists(os.path.join(self.root, self.processed_folder, self.training_file)) and \