"docs/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "61d223c884c1576456a0196f4488f12a16e62206"
Commit 9ff1b4e8 authored by vabh's avatar vabh
Browse files

fix flake8 errors

parent 9958ecbe
......@@ -67,15 +67,15 @@ class SVHN(data.Dataset):
self.data = loaded_mat['X']
# loading from the .mat file gives an np array of type np.uint8
# converting to np.int64, so that we have a LongTensor after
# converting to np.int64, so that we have a LongTensor after
# the conversion from the numpy array
# the squeeze is needed to obtain a 1D tensor
self.labels = loaded_mat['y'].astype(np.int64).squeeze()
# 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]
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))
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