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