b

jameschapman19 · May 10, 2023 · d317829 · d317829
1 parent 4d93055
commit d317829
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/multiviewdata/torchdatasets/mnist.py b/multiviewdata/torchdatasets/mnist.py
@@ -35,8 +35,8 @@ def __len__(self):
 
     def __getitem__(self, idx):
         x_a, label = self.dataset[idx]
-        x_b = x_a[:, :, 14:]
-        x_a = x_a[:, :, :14]
+        x_b = x_a[:, :, 14:]/255.
+        x_a = x_a[:, :, :14]/255.
         if self.flatten:
             x_a = torch.flatten(x_a)
             x_b = torch.flatten(x_b)
@@ -154,7 +154,9 @@ def load_mnist(mnist_type, train, root, download):
             train=train,
             download=download,
             transform=torchvision.transforms.Compose(
-                [torchvision.transforms.ToTensor()]
+                [
+                    torchvision.transforms.ToTensor(),
+                ]
             ),
         )