diff --git a/docs/source/datasets.rst b/docs/source/datasets.rst index 3152a82365c..7fc7c5da1ca 100644 --- a/docs/source/datasets.rst +++ b/docs/source/datasets.rst @@ -86,6 +86,8 @@ CIFAR :members: __getitem__ :special-members: +.. autoclass:: CIFAR100 + STL10 ~~~~~ diff --git a/torchvision/datasets/cifar.py b/torchvision/datasets/cifar.py index 67f15fbdc6d..c3b7b8ef4f2 100644 --- a/torchvision/datasets/cifar.py +++ b/torchvision/datasets/cifar.py @@ -161,6 +161,10 @@ def download(self): class CIFAR100(CIFAR10): + """`CIFAR100 `_ Dataset. + + This is a subclass of the `CIFAR10` Dataset. + """ base_folder = 'cifar-100-python' url = "https://www.cs.toronto.edu/~kriz/cifar-100-python.tar.gz" filename = "cifar-100-python.tar.gz"