fix: fix paths to datasets

Author: dimakis

demo-notebooks/guided-demos/download_mnist_datasets.py

@@ -41,7 +41,6 @@ def download_mnist_dataset(destination_dir):
 
 
 # Specify the directory where you
-script_dir = os.path.dirname(os.path.abspath(__file__))
-destination_dir = script_dir + "/mnist_datasets"
+destination_dir = os.path.dirname(os.path.abspath(__file__))
 
 download_mnist_dataset(destination_dir)

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.dirname(os.path.abspath(__file__) + "/mnist_datasets")
+local_minst_path = os.path.dirname(os.path.abspath(__file__) + "/MNIST/raw")
 
 print("prior to running the trainer")
 print("MASTER_ADDR: is ", os.getenv("MASTER_ADDR"))