refactor: correct path to datasets

dimakis · Nov 14, 2023 · 0ed4cfc · 0ed4cfc
1 parent 55455b1
commit 0ed4cfc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/demo-notebooks/guided-demos/mnist_disconnected.py b/demo-notebooks/guided-demos/mnist_disconnected.py
@@ -30,7 +30,7 @@
 BATCH_SIZE = 256 if torch.cuda.is_available() else 64
 # %%
 
-local_minst_path = os.path.join(PATH_DATASETS, "mnist")
+local_minst_path = os.path.dirname(os.path.abspath(__file__) + "/mnist_datasets")
 
 print("prior to running the trainer")
 print("MASTER_ADDR: is ", os.getenv("MASTER_ADDR"))
@@ -114,7 +114,7 @@ def configure_optimizers(self):
 
     def prepare_data(self):
         # download
-        print("Downloading MNIST dataset...")
+        print("Preparing MNIST dataset...")
         MNIST(self.data_dir, train=True, download=False)
         MNIST(self.data_dir, train=False, download=False)