-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Closed
Labels
Description
If you miss-spell, or otherwise pass an invalid fold to the STL dataset it will set the data and labels to that of the test set.
Can be seen here:
vision/torchvision/datasets/stl10.py
Line 75 in fa377c4
else: # self.split == 'test': |
Other datasets appear to raise a ValueError
in their constructor when passed an invalid fold:
vision/torchvision/datasets/mnist.py
Line 202 in fa377c4
if split not in self.splits: |
Would be good to do so for STL-10 also.