diff --git a/torchvision/datasets/svhn.py b/torchvision/datasets/svhn.py index dd6c42dab36..8a2e5839971 100644 --- a/torchvision/datasets/svhn.py +++ b/torchvision/datasets/svhn.py @@ -19,8 +19,7 @@ class SVHN(VisionDataset): This class needs `scipy `_ to load data from `.mat` format. Args: - root (string): Root directory of dataset where directory - ``SVHN`` exists. + root (string): Root directory of the dataset where the data is stored. split (string): One of {'train', 'test', 'extra'}. Accordingly dataset is selected. 'extra' is Extra training set. transform (callable, optional): A function/transform that takes in an PIL image