Skip to content

Commit d98c886

Browse files
authored
Merge pull request #1612 from d2l-ai/master
Release v0.16.0 fix
2 parents 1c7414e + 69ba3d6 commit d98c886

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

chapter_computer-vision/image-augmentation.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -223,8 +223,8 @@ def load_cifar10(is_train, augs, batch_size):
223223
```{.python .input}
224224
#@tab pytorch
225225
def load_cifar10(is_train, augs, batch_size):
226-
dataset = torchvision.datasets.CIFAR10(
227-
root="../data/cifar10", train=is_train, transform=augs, download=True)
226+
dataset = torchvision.datasets.CIFAR10(root="../data", train=is_train,
227+
transform=augs, download=True)
228228
dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size,
229229
shuffle=is_train, num_workers=d2l.get_dataloader_workers())
230230
return dataloader

0 commit comments

Comments
 (0)