Skip to content

GitLab

Commit 9ff1b4e8 authored by vabh's avatar vabh
Browse files

fix flake8 errors

parent 9958ecbe
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
torchvision/datasets/svhn.py
View file @ 9ff1b4e8
...@@ -75,7 +75,7 @@ class SVHN(data.Dataset): ...@@ -75,7 +75,7 @@ class SVHN(data.Dataset):
# the svhn dataset assigns the class label "10" to the digit 0 # the svhn dataset assigns the class label "10" to the digit 0
# this makes it inconsistent with several loss functions # this makes it inconsistent with several loss functions
# which expect the class labels to be in the range [0, C-1] # which expect the class labels to be in the range [0, C-1]
np.place(self.labels, self.labels==10, 0) np.place(self.labels, self.labels == 10, 0)
self.data = np.transpose(self.data, (3, 2, 0, 1)) self.data = np.transpose(self.data, (3, 2, 0, 1))
def __getitem__(self, index): def __getitem__(self, index):
......
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