Skip to content

GitLab

"examples/pytorch/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "5b2d44faaaea47c76cdf5d721d1f69eb9c7f620a"
Commit 4aff146c authored by Keon's avatar Keon Committed by Soumith Chintala
Browse files

Return real dataset length (#210)

parent 7842c3ed
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/datasets/mnist.py
View file @ 4aff146c
...@@ -81,9 +81,9 @@ class MNIST(data.Dataset): ...@@ -81,9 +81,9 @@ class MNIST(data.Dataset):
def __len__(self): def __len__(self):
if self.train: if self.train:
return 60000 return len(self.train_data)
else: else:
return 10000 return len(self.test_data)
def _check_exists(self): def _check_exists(self):
return os.path.exists(os.path.join(self.root, self.processed_folder, self.training_file)) and \ return os.path.exists(os.path.join(self.root, self.processed_folder, self.training_file)) and \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment