From 6720c7679c4ba668016a56e8bc92a6f0f564412e Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Wed, 21 Sep 2022 12:43:37 -0700 Subject: [PATCH] download cifar to blob storage --- tests/unit/alexnet_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/alexnet_model.py b/tests/unit/alexnet_model.py index ab51a4630b7f..b94bd7052a9f 100755 --- a/tests/unit/alexnet_model.py +++ b/tests/unit/alexnet_model.py @@ -114,7 +114,7 @@ def cifar_trainset(fp16=False): dist.barrier() if local_rank != 0: dist.barrier() - trainset = torchvision.datasets.CIFAR10(root='/tmp/cifar10-data', + trainset = torchvision.datasets.CIFAR10(root='/blob/cifar10-data', train=True, download=True, transform=transform)