Severe Bug Fixed: MNIST and USPS has different scales (0-1) vs (0-255…

…) in dataloader
corenel · Apr 12, 2022 · 13a295a · 13a295a
1 parent 3d38435
commit 13a295a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datasets/usps.py b/datasets/usps.py
@@ -57,7 +57,7 @@ def __init__(self, root, train=True, transform=None, download=False):
             np.random.shuffle(indices)
             self.train_data = self.train_data[indices[0:self.dataset_size], ::]
             self.train_labels = self.train_labels[indices[0:self.dataset_size]]
-        self.train_data *= 255.0
+
         self.train_data = self.train_data.transpose(
             (0, 2, 3, 1))  # convert to HWC