From d1b8632766887dec276f18e4672ea160ac752d6d Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Wed, 18 Oct 2023 14:41:31 -0400 Subject: [PATCH] download cifar10 --- only_for_me/narval/pytorch-ddp-test-pl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/only_for_me/narval/pytorch-ddp-test-pl.py b/only_for_me/narval/pytorch-ddp-test-pl.py index 0c89cd3f..eb35ed85 100644 --- 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)