Download mnist globally under /tmp

Signed-off-by: MasterSkepticista <[email protected]>
securefederatedai · Jan 16, 2025 · b6a9505 · b6a9505
1 parent a110bae
commit b6a9505
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/openfl-workspace/torch_cnn_mnist/src/dataloader.py b/openfl-workspace/torch_cnn_mnist/src/dataloader.py
@@ -121,7 +121,7 @@ def _load_raw_datashards(shard_num, collaborator_count, transform=None):
         2 tuples: (image, label) of the training, validation dataset
     """
     train_data, val_data = (
-        datasets.MNIST("data", train=train, download=True, transform=transform)
+        datasets.MNIST("/tmp/mnist", train=train, download=True, transform=transform)
         for train in (True, False)
     )
     X_train_tot, y_train_tot = train_data.train_data, train_data.train_labels

diff --git a/openfl-workspace/torch_cnn_mnist_fed_eval/src/mnist_utils.py b/openfl-workspace/torch_cnn_mnist_fed_eval/src/mnist_utils.py
@@ -41,7 +41,7 @@ def _load_raw_datashards(shard_num, collaborator_count, transform=None):
         2 tuples: (image, label) of the training, validation dataset
     """
     train_data, val_data = (
-        datasets.MNIST('data', train=train, download=True, transform=transform)
+        datasets.MNIST('/tmp/mnist', train=train, download=True, transform=transform)
         for train in (True, False)
     )
     X_train_tot, y_train_tot = train_data.train_data, train_data.train_labels

diff --git a/openfl-workspace/torch_cnn_mnist_straggler_check/src/mnist_utils.py b/openfl-workspace/torch_cnn_mnist_straggler_check/src/mnist_utils.py
@@ -41,7 +41,7 @@ def _load_raw_datashards(shard_num, collaborator_count, transform=None):
         2 tuples: (image, label) of the training, validation dataset
     """
     train_data, val_data = (
-        datasets.MNIST('data', train=train, download=True, transform=transform)
+        datasets.MNIST('/tmp/mnist', train=train, download=True, transform=transform)
         for train in (True, False)
     )
     X_train_tot, y_train_tot = train_data.train_data, train_data.train_labels