arthur/ml_exercises
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/ml_exercises.git synced 2026-01-14 12:14:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

minor

Browse Source
This commit is contained in:
franzi
2022-02-26 14:05:13 +01:00
parent c9244982d1
commit 69119b140c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
notebooks/5_mnist_torch.ipynb
View File

@@ -182,7 +182,7 @@
"outputs": [],
"source": [
"# apply network to input, i.e., call forward() to generate the prediction\n",
"y = ffnn(x)\n",
"y = ffnn(x) # same as: y = ffnn.forward(x)\n",
"print(y.shape) # batch size x classes"
]
},
@@ -667,7 +667,7 @@
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
@@ -681,7 +681,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.9.5"
"version": "3.8.5"
}
},
"nbformat": 4,