b

jameschapman19 · May 10, 2023 · 077c99b · 077c99b
1 parent d317829
commit 077c99b
Showing 1 changed file with 2 additions and 2 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:]/255.
-        x_a = x_a[:, :, :14]/255.
+        x_b = x_a[:, :, 14:]
+        x_a = x_a[:, :, :14]
         if self.flatten:
             x_a = torch.flatten(x_a)
             x_b = torch.flatten(x_b)