From 69ba3d67a09a429d0402f950feb1c72b00ea0a8d Mon Sep 17 00:00:00 2001
From: Aston Zhang <asv325@gmail.com>
Date: Wed, 6 Jan 2021 07:21:31 +0000
Subject: [PATCH] pytorch cifar data root

---
 chapter_computer-vision/image-augmentation.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/chapter_computer-vision/image-augmentation.md b/chapter_computer-vision/image-augmentation.md
index d25ea85dc5..a15ca3f37b 100644
--- 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