Cool everything working!

jameschapman19 · Mar 3, 2022 · fc0f13e · fc0f13e
1 parent 5fb7524
commit fc0f13e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/multiviewdata/torchdatasets/mnist.py b/multiviewdata/torchdatasets/mnist.py
@@ -99,7 +99,7 @@ def __getitem__(self, index):
         if self.flatten:
             x_a = torch.flatten(x_a)
             x_b = torch.flatten(x_b)
-        return {"views": (x_a.astype(np.float32), x_b.astype(np.float32)), "label": label.astype(np.float32), "index": index}
+        return {"views": (x_a.float(), x_b.float()), "label": label, "index": index}
 
 
 class TangledMNISTDataset(Dataset):