Skip to content

Commit 4e61dc4

Browse files
authored
Fix d_out code comment in bonus materials (rasbt#715)
1 parent c4ec55e commit 4e61dc4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ch03/02_bonus_efficient-multihead-attention/mha-implementations.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -689,7 +689,7 @@
689689
" def __init__(self, d_in, d_out, num_heads, context_length, dropout=0.0, qkv_bias=False):\n",
690690
" super().__init__()\n",
691691
"\n",
692-
" assert d_out % num_heads == 0, \"embed_dim is indivisible by num_heads\"\n",
692+
" assert d_out % num_heads == 0, \"d_out is indivisible by num_heads\"\n",
693693
"\n",
694694
" self.num_heads = num_heads\n",
695695
" self.context_length = context_length\n",
@@ -1002,7 +1002,7 @@
10021002
" def __init__(self, d_in, d_out, num_heads, context_length, dropout=0.0, qkv_bias=False):\n",
10031003
" super().__init__()\n",
10041004
"\n",
1005-
" assert d_out % num_heads == 0, \"embed_dim is indivisible by num_heads\"\n",
1005+
" assert d_out % num_heads == 0, \"d_out is indivisible by num_heads\"\n",
10061006
"\n",
10071007
" self.num_heads = num_heads\n",
10081008
" self.context_length = context_length\n",

0 commit comments

Comments
 (0)