dr_domi/udlbook
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(8.1) : error in Chap08\8_1_MNIST_1D_Performance.ipynb

Browse Source
This commit is contained in:
muddlebee
2024-08-17 03:50:15 +05:30
parent 3c8dab14e6
commit b423a67855
1 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Notebooks/Chap08/8_1_MNIST_1D_Performance.ipynb
View File

@@ -117,20 +117,11 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 12, "execution_count": null,
"metadata": { "metadata": {
"id": "_rX6N3VyyQTY" "id": "_rX6N3VyyQTY"
}, },
"outputs": [ "outputs": [],
{
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 24, train loss 1.584953, train error 62.60, test loss 1.665801, test error 67.90\n",
"Epoch 25, train loss 1.586464, train error 63.05, test loss 1.666717, test error 68.00\n"
]
}
],
"source": [ "source": [
"# choose cross entropy loss function (equation 5.24)\n", "# choose cross entropy loss function (equation 5.24)\n",
"loss_function = torch.nn.CrossEntropyLoss()\n", "loss_function = torch.nn.CrossEntropyLoss()\n",