From 13a295ab5c94a572854cdf0ffa5e63c65e209777 Mon Sep 17 00:00:00 2001 From: Yuhui Zhang Date: Mon, 11 Apr 2022 20:03:33 -0700 Subject: [PATCH] Severe Bug Fixed: MNIST and USPS has different scales (0-1) vs (0-255) in dataloader --- datasets/usps.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/usps.py b/datasets/usps.py index 62c4629..f6d906e 100644 --- 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