Simplify embedding vs linear layer code (#278)

2025-09-23 07:05:27 +00:00 · 2024-07-21 10:21:10 -07:00 · 2024-07-21 10:21:10 -07:00 · fa56c80402
commit fa56c80402
parent bbe09e9799
1 changed files with 2 additions and 2 deletions
--- a/ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb
+++ b/ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb
@ -47,7 +47,7 @@
     "name": "stdout",
     "output_type": "stream",
     "text": [
-      "PyTorch version: 2.1.0\n"
+      "PyTorch version: 2.3.1\n"
     ]
    }
   ],
@ -373,7 +373,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "linear.weight = torch.nn.Parameter(embedding.weight.T.detach())"
+    "linear.weight = torch.nn.Parameter(embedding.weight.T)"
   ]
  },
  {