Skip to content

Commit 5db8e93

Browse files
committed
loss backward spotlight
1 parent 4092fe4 commit 5db8e93

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lab2/solutions/PT_Part1_MNIST_Solution.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -658,7 +658,7 @@
658658
"source": [
659659
"### Train and test the CNN model\n",
660660
"\n",
661-
"Earlier in the lab, we defined a `train` function. The body of the function is quite useful because it allows us to have control over the training model, and to record differentiation operations during training by computing the gradients using loss.backward(). You may recall seeing this in Lab 1 Part 1.\n",
661+
"Earlier in the lab, we defined a `train` function. The body of the function is quite useful because it allows us to have control over the training model, and to record differentiation operations during training by computing the gradients using `loss.backward()`. You may recall seeing this in Lab 1 Part 1.\n",
662662
"\n",
663663
"We'll use this same framework to train our `cnn_model` using stochastic gradient descent. You are free to implement the following parts with or without the train and evaluate functions we defined above. What is most important is understanding how to manipulate the bodies of those functions to train and test models. \n",
664664
"\n",

0 commit comments

Comments
 (0)