Commit 81ca2a1d authored by Philip Meier's avatar Philip Meier Committed by Soumith Chintala
Browse files

add documentation for ImageNet dataset (#859)

parent 07a02460
...@@ -7,7 +7,7 @@ Hence, they can all be passed to a :class:`torch.utils.data.DataLoader` ...@@ -7,7 +7,7 @@ Hence, they can all be passed to a :class:`torch.utils.data.DataLoader`
which can load multiple samples parallelly using ``torch.multiprocessing`` workers. which can load multiple samples parallelly using ``torch.multiprocessing`` workers.
For example: :: For example: ::
imagenet_data = torchvision.datasets.ImageFolder('path/to/imagenet_root/') imagenet_data = torchvision.datasets.ImageNet('path/to/imagenet_root/')
data_loader = torch.utils.data.DataLoader(imagenet_data, data_loader = torch.utils.data.DataLoader(imagenet_data,
batch_size=4, batch_size=4,
shuffle=True, shuffle=True,
...@@ -100,12 +100,11 @@ DatasetFolder ...@@ -100,12 +100,11 @@ DatasetFolder
Imagenet-12 Imagenet-12
~~~~~~~~~~~ ~~~~~~~~~~~
This should simply be implemented with an ``ImageFolder`` dataset. .. autoclass:: ImageNet
The data is preprocessed `as described
here <https://github.com/facebook/fb.resnet.torch/blob/master/INSTALL.md#download-the-imagenet-dataset>`__ .. note ::
This requires `scipy` to be installed
`Here is an
example <https://github.com/pytorch/examples/blob/e0d33a69bec3eb4096c265451dbb85975eb961ea/imagenet/main.py#L113-L126>`__.
CIFAR CIFAR
~~~~~ ~~~~~
......
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