Merge pull request #1612 from d2l-ai/master

Release v0.16.0 fix
d2l-ai · Jan 6, 2021 · d98c886 · d98c886
2 parents 1c7414e + 69ba3d6
commit d98c886
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chapter_computer-vision/image-augmentation.md b/chapter_computer-vision/image-augmentation.md
@@ -223,8 +223,8 @@ def load_cifar10(is_train, augs, batch_size):
 ```{.python .input}
 #@tab pytorch
 def load_cifar10(is_train, augs, batch_size):
-    dataset = torchvision.datasets.CIFAR10(
-        root="../data/cifar10", train=is_train, transform=augs, download=True)
+    dataset = torchvision.datasets.CIFAR10(root="../data", train=is_train,
+                                           transform=augs, download=True)
     dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size,
                     shuffle=is_train, num_workers=d2l.get_dataloader_workers())
     return dataloader