Merge pull request #477 from ggraffieti/master

Dataset download only in ~/.avalanche/data folder

Author: vlomonaco

examples/confusion_matrix.py

@@ -17,6 +17,8 @@
 from __future__ import division
 from __future__ import print_function
 
+from os.path import expanduser
+
 import argparse
 import torch
 from torch.nn import CrossEntropyLoss
@@ -53,10 +55,10 @@ def main(args):
     # ---------
 
     # --- SCENARIO CREATION
-    mnist_train = MNIST('./data/mnist', train=True,
-                        download=True, transform=train_transform)
-    mnist_test = MNIST('./data/mnist', train=False,
-                       download=True, transform=test_transform)
+    mnist_train = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                        train=True, download=True, transform=train_transform)
+    mnist_test = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                       train=False, download=True, transform=test_transform)
     scenario = nc_scenario(
         mnist_train, mnist_test, 5, task_labels=False, seed=1234)
     # ---------

examples/eval_plugin.py

@@ -17,6 +17,8 @@
 from __future__ import division
 from __future__ import print_function
 
+from os.path import expanduser
+
 import argparse
 import torch
 from torch.nn import CrossEntropyLoss
@@ -55,10 +57,10 @@ def main(args):
     # ---------
 
     # --- SCENARIO CREATION
-    mnist_train = MNIST('./data/mnist', train=True,
-                        download=True, transform=train_transform)
-    mnist_test = MNIST('./data/mnist', train=False,
-                       download=True, transform=test_transform)
+    mnist_train = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                        train=True, download=True, transform=train_transform)
+    mnist_test = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                       train=False, download=True, transform=test_transform)
     scenario = nc_scenario(
         mnist_train, mnist_test, 5, task_labels=False, seed=1234)
     # ---------

examples/getting_started.py

@@ -17,6 +17,8 @@
 from __future__ import division
 from __future__ import print_function
 
+from os.path import expanduser
+
 import argparse
 import torch
 from torch.nn import CrossEntropyLoss
@@ -50,10 +52,10 @@ def main(args):
     # ---------
 
     # --- SCENARIO CREATION
-    mnist_train = MNIST('./data/mnist', train=True,
-                        download=True, transform=train_transform)
-    mnist_test = MNIST('./data/mnist', train=False,
-                       download=True, transform=test_transform)
+    mnist_train = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                        train=True, download=True, transform=train_transform)
+    mnist_test = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                       train=False, download=True, transform=test_transform)
     scenario = nc_scenario(
         mnist_train, mnist_test, 5, task_labels=False, seed=1234)
     # ---------

examples/getting_started_no_avalanche.py

@@ -18,6 +18,8 @@
 from __future__ import division
 from __future__ import print_function
 
+from os.path import expanduser
+
 import argparse
 import torch
 import torch.nn as nn
@@ -95,10 +97,10 @@ def forward(self, x):
         new_test_transform = transforms.Compose(test_transform_list)
 
         # get the datasets with the constructed transformation
-        permuted_train = MNIST(root='./data/mnist',
+        permuted_train = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
                                train=True,
                                download=True, transform=new_train_transform)
-        permuted_test = MNIST(root='./data/mnist',
+        permuted_test = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
                               train=False,
                               download=True, transform=new_test_transform)
 

examples/replay.py

@@ -17,6 +17,8 @@
 from __future__ import division
 from __future__ import print_function
 
+from os.path import expanduser
+
 import argparse
 import torch
 from torch.nn import CrossEntropyLoss
@@ -56,10 +58,10 @@ def main(args):
     # ---------
 
     # --- SCENARIO CREATION
-    mnist_train = MNIST('./data/mnist', train=True,
-                        download=True, transform=train_transform)
-    mnist_test = MNIST('./data/mnist', train=False,
-                       download=True, transform=test_transform)
+    mnist_train = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                        train=True, download=True, transform=train_transform)
+    mnist_test = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                       train=False, download=True, transform=test_transform)
     scenario = nc_scenario(
         mnist_train, mnist_test, n_batches, task_labels=False, seed=1234)
     # ---------

examples/tensorboard_logger.py

@@ -18,6 +18,8 @@
 from __future__ import division
 from __future__ import print_function
 
+from os.path import expanduser
+
 import argparse
 import torch
 from torch.nn import CrossEntropyLoss
@@ -56,10 +58,10 @@ def main(args):
     # ---------
 
     # --- SCENARIO CREATION
-    mnist_train = MNIST('./data/mnist', train=True,
-                        download=True, transform=train_transform)
-    mnist_test = MNIST('./data/mnist', train=False,
-                       download=True, transform=test_transform)
+    mnist_train = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                        train=True, download=True, transform=train_transform)
+    mnist_test = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/",
+                       train=False, download=True, transform=test_transform)
     scenario = nc_scenario(
         mnist_train, mnist_test, 5, task_labels=False, seed=1234)
     # ---------