download cifar10

mwalmsley · Oct 18, 2023 · d1b8632 · d1b8632
1 parent 516a569
commit d1b8632
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/only_for_me/narval/pytorch-ddp-test-pl.py b/only_for_me/narval/pytorch-ddp-test-pl.py
@@ -67,7 +67,7 @@ def configure_optimizers(self):
 
     transform_train = transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
 
-    dataset_train = CIFAR10(root='./data', train=True, download=False, transform=transform_train)
+    dataset_train = CIFAR10(root='/project/def-bovy/walml/data/roots/cifar10', train=True, download=False, transform=transform_train)
 
     train_loader = DataLoader(dataset_train, batch_size=args.batch_size, num_workers=args.num_workers)