From 176f1458cc0c1525ccc3594527fb649607085014 Mon Sep 17 00:00:00 2001 From: Daniel Angelov Date: Mon, 2 May 2022 18:07:20 +0300 Subject: [PATCH] Update SVHN docs --- torchvision/datasets/svhn.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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