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

This commit is contained in:
muddlebee
2024-08-17 03:50:15 +05:30
parent 3c8dab14e6
commit b423a67855

View File

@@ -117,20 +117,11 @@
},
{
"cell_type": "code",
"execution_count": 12,
"execution_count": null,
"metadata": {
"id": "_rX6N3VyyQTY"
},
"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"
]
}
],
"outputs": [],
"source": [
"# choose cross entropy loss function (equation 5.24)\n",
"loss_function = torch.nn.CrossEntropyLoss()\n",