Эх сурвалжийг харах

Simplify embedding vs linear layer code (#278)

Sebastian Raschka 1 жил өмнө
parent
commit
fa56c80402

+ 2 - 2
ch02/03_bonus_embedding-vs-matmul/embeddings-and-linear-layers.ipynb

@@ -47,7 +47,7 @@
      "name": "stdout",
      "name": "stdout",
      "output_type": "stream",
      "output_type": "stream",
      "text": [
      "text": [
-      "PyTorch version: 2.1.0\n"
+      "PyTorch version: 2.3.1\n"
      ]
      ]
     }
     }
    ],
    ],
@@ -373,7 +373,7 @@
    "metadata": {},
    "metadata": {},
    "outputs": [],
    "outputs": [],
    "source": [
    "source": [
-    "linear.weight = torch.nn.Parameter(embedding.weight.T.detach())"
+    "linear.weight = torch.nn.Parameter(embedding.weight.T)"
    ]
    ]
   },
   },
   {
   {