fix(8.1) : error in Chap08\8_1_MNIST_1D_Performance.ipynb
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user