Skip to content

Commit f7eab40

Browse files
committed
Make sure we're loading MNIST from the same spot in Parts 2 and 3.
1 parent c53ca4a commit f7eab40

File tree

3 files changed

+10
-2
lines changed

3 files changed

+10
-2
lines changed

intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@
5858
" ])\n",
5959
"\n",
6060
"# Download and load the training data\n",
61-
"trainset = datasets.MNIST('MNIST_data/', download=True, train=True, transform=transform)\n",
61+
"trainset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=True, transform=transform)\n",
6262
"trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)"
6363
]
6464
},

intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@
5757
" transforms.Normalize((0.5,), (0.5,)),\n",
5858
" ])\n",
5959
"# Download and load the training data\n",
60-
"trainset = datasets.MNIST('MNIST_data/', download=True, train=True, transform=transform)\n",
60+
"trainset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=True, transform=transform)\n",
6161
"trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)"
6262
]
6363
},

intro-to-pytorch/Part 3 - Training Neural Networks (Exercises).ipynb

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,14 @@
9393
"trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)"
9494
]
9595
},
96+
{
97+
"cell_type": "markdown",
98+
"metadata": {},
99+
"source": [
100+
"### Note\n",
101+
"If you haven't seen `nn.Sequential` yet, please finish the end of the Part 2 notebook."
102+
]
103+
},
96104
{
97105
"cell_type": "code",
98106
"execution_count": null,

0 commit comments

Comments
 (0)