Skip to content

Commit 8c9ffbe

Browse files
committed
Fix a comment in the Part 3 notebook
1 parent ae5248a commit 8c9ffbe

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@
465465
"with torch.no_grad():\n",
466466
" logps = model(img)\n",
467467
"\n",
468-
"# Output of the network are logits, need to take softmax for probabilities\n",
468+
"# Output of the network are log-probabilities, need to take exponential for probabilities\n",
469469
"ps = torch.exp(logps)\n",
470470
"helper.view_classify(img.view(1, 28, 28), ps)"
471471
]

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -627,7 +627,7 @@
627627
"with torch.no_grad():\n",
628628
" logps = model(img)\n",
629629
"\n",
630-
"# Output of the network are logits, need to take softmax for probabilities\n",
630+
"# Output of the network are log-probabilities, need to take exponential for probabilities\n",
631631
"ps = torch.exp(logps)\n",
632632
"helper.view_classify(img.view(1, 28, 28), ps)"
633633
]

0 commit comments

Comments
 (0)