Skip to content

Commit

Permalink
updated
Browse files Browse the repository at this point in the history
  • Loading branch information
arun477 committed Sep 14, 2023
1 parent 19d498a commit b85505a
Showing 1 changed file with 23 additions and 6 deletions.
29 changes: 23 additions & 6 deletions mnist_classifier.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -180,7 +180,24 @@
"cell_type": "code",
"execution_count": 109,
"metadata": {},
"outputs": [],
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"train, epoch:1, loss: 0.1077, accuracy: 0.9104\n",
"eval, epoch:1, loss: 0.0382, accuracy: 0.9791\n",
"train, epoch:2, loss: 0.0410, accuracy: 0.9832\n",
"eval, epoch:2, loss: 0.0221, accuracy: 0.9866\n",
"train, epoch:3, loss: 0.0538, accuracy: 0.9871\n",
"eval, epoch:3, loss: 0.0141, accuracy: 0.9887\n",
"train, epoch:4, loss: 0.0343, accuracy: 0.9858\n",
"eval, epoch:4, loss: 0.0163, accuracy: 0.9871\n",
"train, epoch:5, loss: 0.0390, accuracy: 0.9865\n",
"eval, epoch:5, loss: 0.0169, accuracy: 0.9871\n"
]
}
],
"source": [
"model = cnn_classifier()\n",
"model.apply(kaiming_init)\n",
Expand Down Expand Up @@ -210,16 +227,16 @@
},
{
"cell_type": "code",
"execution_count": 95,
"execution_count": 110,
"metadata": {
"tags": [
"exclude"
]
},
"outputs": [],
"source": [
"# with open('./classifier.pkl', 'wb') as model_file:\n",
"# pickle.dump(model, model_file)"
"with open('./classifier.pkl', 'wb') as model_file:\n",
" pickle.dump(model, model_file)"
]
},
{
Expand All @@ -235,7 +252,7 @@
},
{
"cell_type": "code",
"execution_count": 96,
"execution_count": 111,
"metadata": {
"tags": [
"exclude"
Expand All @@ -247,7 +264,7 @@
"output_type": "stream",
"text": [
"[NbConvertApp] Converting notebook mnist_classifier.ipynb to script\n",
"[NbConvertApp] Writing 5934 bytes to mnist_classifier.py\n"
"[NbConvertApp] Writing 3691 bytes to mnist_classifier.py\n"
]
}
],
Expand Down

0 comments on commit b85505a

Please sign in to comment.