fix issue with minist dataloader

Author: Hananel-Hazan

examples/mnist/SOM_LM-SNNs.py

@@ -77,7 +77,7 @@
 
 # Determines number of workers to use
 if n_workers == -1:
-    n_workers = torch.cuda.is_available() * 4 * torch.cuda.device_count()
+    n_workers = 0  # torch.cuda.is_available() * 4 * torch.cuda.device_count()
 
 n_sqrt = int(np.ceil(np.sqrt(n_neurons)))
 start_intensity = intensity

examples/mnist/batch_eth_mnist.py

@@ -49,7 +49,7 @@
 parser.add_argument("--test", dest="train", action="store_false")
 parser.add_argument("--plot", dest="plot", action="store_true")
 parser.add_argument("--gpu", dest="gpu", action="store_true")
-parser.set_defaults(plot=True, gpu=False)
+parser.set_defaults(plot=True, gpu=True)
 
 args = parser.parse_args()
 
@@ -90,7 +90,7 @@
 
 # Determines number of workers to use
 if n_workers == -1:
-    n_workers = gpu * 4 * torch.cuda.device_count()
+    n_workers = 0  # gpu * 1 * torch.cuda.device_count()
 
 n_sqrt = int(np.ceil(np.sqrt(n_neurons)))
 start_intensity = intensity
@@ -116,7 +116,7 @@
 dataset = MNIST(
     PoissonEncoder(time=time, dt=dt),
     None,
-    root=os.path.join(ROOT_DIR, "data", "MNIST"),
+    "../../data/MNIST",
     download=True,
     transform=transforms.Compose(
         [transforms.ToTensor(), transforms.Lambda(lambda x: x * intensity)]

examples/mnist/conv_mnist.py

@@ -41,7 +41,7 @@
 parser.add_argument("--test", dest="train", action="store_false")
 parser.add_argument("--plot", dest="plot", action="store_true")
 parser.add_argument("--gpu", dest="gpu", action="store_true")
-parser.set_defaults(plot=True, gpu=False, train=True)
+parser.set_defaults(plot=True, gpu=True, train=True)
 
 args = parser.parse_args()
 
@@ -164,7 +164,7 @@
         start = t()
 
     train_dataloader = torch.utils.data.DataLoader(
-        train_dataset, batch_size=1, shuffle=True, num_workers=4, pin_memory=gpu
+        train_dataset, batch_size=1, shuffle=True, num_workers=0, pin_memory=gpu
     )
 
     for step, batch in enumerate(tqdm(train_dataloader)):

examples/mnist/eth_mnist.py

@@ -48,7 +48,7 @@
 parser.add_argument("--test", dest="train", action="store_false")
 parser.add_argument("--plot", dest="plot", action="store_true")
 parser.add_argument("--gpu", dest="gpu", action="store_true")
-parser.set_defaults(plot=True, gpu=False)
+parser.set_defaults(plot=True, gpu=True)
 
 args = parser.parse_args()
 
@@ -85,7 +85,7 @@
 
 # Determines number of workers to use
 if n_workers == -1:
-    n_workers = gpu * 4 * torch.cuda.device_count()
+    n_workers = 0  # gpu * 4 * torch.cuda.device_count()
 
 if not train:
     update_interval = n_test

examples/mnist/supervised_mnist.py

@@ -46,7 +46,7 @@
 parser.add_argument("--plot", dest="plot", action="store_true")
 parser.add_argument("--gpu", dest="gpu", action="store_true")
 parser.add_argument("--device_id", type=int, default=0)
-parser.set_defaults(plot=True, gpu=False, train=True)
+parser.set_defaults(plot=True, gpu=True, train=True)
 
 args = parser.parse_args()