From 89d3adb19a2ffd583188157e874a87da440fb843 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Thu, 3 Feb 2022 11:16:12 +0000 Subject: [PATCH] Categorize dataset doc page into sections --- docs/source/datasets.rst | 90 ++++++++++++++++++++++++++++++---------- 1 file changed, 68 insertions(+), 22 deletions(-) diff --git a/docs/source/datasets.rst b/docs/source/datasets.rst index 9e5a30fa8ce..717142bb22d 100644 --- a/docs/source/datasets.rst +++ b/docs/source/datasets.rst @@ -5,7 +5,7 @@ Torchvision provides many built-in datasets in the ``torchvision.datasets`` module, as well as utility classes for building your own datasets. Built-in datasets -~~~~~~~~~~~~~~~~~ +----------------- All datasets are subclasses of :class:`torch.utils.data.Dataset` i.e, they have ``__getitem__`` and ``__len__`` methods implemented. @@ -25,6 +25,8 @@ All the datasets have almost similar API. They all have two common arguments: ``transform`` and ``target_transform`` to transform the input and target respectively. You can also create your own datasets using the provided :ref:`base classes `. +Image classification +~~~~~~~~~~~~~~~~~~~~ .. autosummary:: :toctree: generated/ @@ -35,9 +37,6 @@ You can also create your own datasets using the provided :ref:`base classes