fix: removed softmax in random code.

Author: Shamik-07

random_code.ipynb

@@ -588,15 +588,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "835b3357-3368-4944-9411-c10d3a507236",
    "metadata": {},
    "outputs": [],
    "source": [
     "# Define the linear classifier model\n",
     "model = torch.nn.Sequential(\n",
     "    torch.nn.Linear(in_features=4, out_features=3),\n",
-    "    torch.nn.Softmax(dim=1)\n",
+    "    # torch.nn.Softmax(dim=1) # softmax already applied with crossentropy loss in pytorch\n",
     ")"
    ]
   },