variable name fix

zhangjiashen · Feb 10, 2024 · baa8617 · baa8617
1 parent 496b52f
commit baa8617
Showing 1 changed file with 16 additions and 14 deletions.
diff --git a/ch04/01_main-chapter-code/ch04.ipynb b/ch04/01_main-chapter-code/ch04.ipynb
@@ -185,15 +185,12 @@
    "metadata": {},
    "outputs": [
     {
-     "data": {
-      "text/plain": [
-       "tensor([[ 6109,  3626,  6100,   345,  2651,    13],\n",
-       "        [ 6109,  1110,  6622,   257, 11483,    13]])"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "tensor([[ 6109,  3626,  6100,   345,  2651,    13],\n",
+      "        [ 6109,  1110,  6622,   257, 11483,    13]])\n"
+     ]
     }
    ],
    "source": [
@@ -680,7 +677,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 30,
+   "execution_count": 16,
    "id": "928e7f7c-d0b1-499f-8d07-4cadb428a6f9",
    "metadata": {},
    "outputs": [
@@ -741,7 +738,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 38,
+   "execution_count": 17,
    "id": "05473938-799c-49fd-86d4-8ed65f94fee6",
    "metadata": {},
    "outputs": [
@@ -977,14 +974,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 31,
    "id": "252b78c2-4404-483b-84fe-a412e55c16fc",
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "Input batch:\n",
+      " tensor([[ 6109,  3626,  6100,   345,  2651,    13],\n",
+      "        [ 6109,  1110,  6622,   257, 11483,    13]])\n",
+      "\n",
       "Output shape: torch.Size([2, 6, 50257])\n",
       "tensor([[[ 0.2237,  0.1153,  0.1121,  ...,  0.1412, -0.0542, -0.3782],\n",
       "         [ 0.5285, -0.0155, -0.5074,  ..., -0.3225,  0.4875, -0.0612],\n",
@@ -1005,10 +1006,11 @@
    ],
    "source": [
     "torch.manual_seed(123)\n",
-    "model = GPT(GPT_CONFIG_124M)\n",
+    "model = GPTModel(GPT_CONFIG_124M)\n",
     "\n",
     "out = model(batch)\n",
-    "print(\"Output shape:\", out.shape)\n",
+    "print(\"Input batch:\\n\", batch)\n",
+    "print(\"\\nOutput shape:\", out.shape)\n",
     "print(out)"
    ]
   },