update data directory

Author: orion-orion

optimization/asgd.py

@@ -175,7 +175,7 @@ def run_ps(trainers):
     transforms.ToTensor(),
     transforms.Normalize((0.1307,), (0.3081,))
     ])
-    train_dataset = datasets.MNIST('./data', train=True, download=True,
+    train_dataset = datasets.MNIST('../data', train=True, download=True,
                        transform=transform)
     local_train_datasets = dataset_split(train_dataset, n_workers)    
 

optimization/ssgd_pytorch.py

@@ -146,9 +146,9 @@ def test(epoch, model, device, dataset, dataloader_kwargs):
         transforms.ToTensor(),
         transforms.Normalize((0.1307,), (0.3081,))
         ])
-    train_dataset = datasets.MNIST('./data', train=True, download=True,
+    train_dataset = datasets.MNIST('../data', train=True, download=True,
                        transform=transform)
-    test_dataset = datasets.MNIST('./data', train=False, download=True,
+    test_dataset = datasets.MNIST('../data', train=False, download=True,
                        transform=transform)
     local_train_datasets = dataset_split(train_dataset, n_workers)