From 5bf8518f322350ba44d444250cc04752029cf149 Mon Sep 17 00:00:00 2001 From: maria8899 Date: Thu, 7 Mar 2019 20:47:21 +0000 Subject: [PATCH] Update directory documentation --- torchvision/datasets/mnist.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/torchvision/datasets/mnist.py b/torchvision/datasets/mnist.py index 1ae37db6f6d..c19157088f2 100644 --- a/torchvision/datasets/mnist.py +++ b/torchvision/datasets/mnist.py @@ -15,8 +15,8 @@ class MNIST(data.Dataset): """`MNIST `_ Dataset. Args: - root (string): Root directory of dataset where ``processed/training.pt`` - and ``processed/test.pt`` exist. + root (string): Root directory of dataset where ``MNIST/processed/training.pt`` + and ``MNIST/processed/test.pt`` exist. train (bool, optional): If True, creates dataset from ``training.pt``, otherwise from ``test.pt``. download (bool, optional): If true, downloads the dataset from the internet and @@ -178,8 +178,8 @@ class FashionMNIST(MNIST): """`Fashion-MNIST `_ Dataset. Args: - root (string): Root directory of dataset where ``processed/training.pt`` - and ``processed/test.pt`` exist. + root (string): Root directory of dataset where ``Fashion-MNIST/processed/training.pt`` + and ``Fashion-MNIST/processed/test.pt`` exist. train (bool, optional): If True, creates dataset from ``training.pt``, otherwise from ``test.pt``. download (bool, optional): If true, downloads the dataset from the internet and @@ -204,8 +204,8 @@ class KMNIST(MNIST): """`Kuzushiji-MNIST `_ Dataset. Args: - root (string): Root directory of dataset where ``processed/training.pt`` - and ``processed/test.pt`` exist. + root (string): Root directory of dataset where ``KMNIST/processed/training.pt`` + and ``KMNIST/processed/test.pt`` exist. train (bool, optional): If True, creates dataset from ``training.pt``, otherwise from ``test.pt``. download (bool, optional): If true, downloads the dataset from the internet and @@ -229,8 +229,8 @@ class EMNIST(MNIST): """`EMNIST `_ Dataset. Args: - root (string): Root directory of dataset where ``processed/training.pt`` - and ``processed/test.pt`` exist. + root (string): Root directory of dataset where ``EMNIST/processed/training.pt`` + and ``EMNIST/processed/test.pt`` exist. split (string): The dataset has 6 different splits: ``byclass``, ``bymerge``, ``balanced``, ``letters``, ``digits`` and ``mnist``. This argument specifies which one to use.