From 352b83d225a99d46af3b47a3790a8c51364ae363 Mon Sep 17 00:00:00 2001 From: rasbt Date: Sun, 11 Feb 2024 08:42:21 -0600 Subject: [PATCH] make softmax explicit --- ch04/01_main-chapter-code/ch04.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ch04/01_main-chapter-code/ch04.ipynb b/ch04/01_main-chapter-code/ch04.ipynb index 81f931d..5001926 100644 --- a/ch04/01_main-chapter-code/ch04.ipynb +++ b/ch04/01_main-chapter-code/ch04.ipynb @@ -1235,7 +1235,7 @@ " # Apply softmax to get probabilities\n", " probas = torch.softmax(logits, dim=-1) # (batch, vocab_size)\n", "\n", - " # Get the idx of the vocab entry with the highest logits value\n", + " # Get the idx of the vocab entry with the highest probability value\n", " idx_next = torch.argmax(probas, dim=-1, keepdim=True) # (batch, 1)\n", "\n", " # Append sampled index to the running sequence\n",