Fix TensorMNIST download folder

ContinualAI · Feb 7, 2024 · 5fbb066 · 5fbb066
1 parent 700968e
commit 5fbb066
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/avalanche/benchmarks/datasets/external_datasets/mnist.py b/avalanche/benchmarks/datasets/external_datasets/mnist.py
@@ -1,4 +1,5 @@
 import dill
+import os
 from torchvision.datasets import MNIST
 from avalanche.benchmarks.datasets import default_dataset_location
 from avalanche.checkpointing import constructor_based_serialization
@@ -23,6 +24,15 @@ def __getitem__(self, index: int):
             target = self.target_transform(target)
 
         return img, target
+
+    # Override property raw_folder
+    @property
+    def raw_folder(self) -> str:
+        return os.path.join(self.root, "MNIST", "raw")
+
+    @property
+    def processed_folder(self) -> str:
+        return os.path.join(self.root, "MNIST", "processed")
 
 
 def get_mnist_dataset(dataset_root):