Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
443c3079
Commit
443c3079
authored
Sep 21, 2017
by
Alykhan Tejani
Committed by
GitHub
Sep 21, 2017
Browse files
Update docstring for svhn (#267)
update SVHN docs to note label change for digit 0
parent
918fdffd
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
0 deletions
+7
-0
README.rst
README.rst
+4
-0
torchvision/datasets/svhn.py
torchvision/datasets/svhn.py
+3
-0
No files found.
README.rst
View file @
443c3079
...
...
@@ -176,6 +176,10 @@ STL10
SVHN
~~~~
Note: The SVHN dataset assigns the label `10` to the digit `0`. However, in this Dataset,
we assign the label `0` to the digit `0` to be compatible with PyTorch loss functions which
expect the class labels to be in the range `[0, C-1]`
``dset.SVHN(root, split='train', transform=None, target_transform=None, download=False)``
- ``root`` : root directory of dataset where there is folder ``SVHN``
...
...
torchvision/datasets/svhn.py
View file @
443c3079
...
...
@@ -9,6 +9,9 @@ from .utils import download_url, check_integrity
class
SVHN
(
data
.
Dataset
):
"""`SVHN <http://ufldl.stanford.edu/housenumbers/>`_ Dataset.
Note: The SVHN dataset assigns the label `10` to the digit `0`. However, in this Dataset,
we assign the label `0` to the digit `0` to be compatible with PyTorch loss functions which
expect the class labels to be in the range `[0, C-1]`
Args:
root (string): Root directory of dataset where directory
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment