Heinrich Dinkel commited on
Commit ·
5340ead
1
Parent(s): 0f35cd3
updated
Browse files- notebook.ipynb +1 -1
notebook.ipynb
CHANGED
|
@@ -214,7 +214,7 @@
|
|
| 214 |
" classifier.to(device)\n",
|
| 215 |
" \n",
|
| 216 |
" # Training setup\n",
|
| 217 |
-
" optimizer = torch.optim.Adam(classifier.parameters(), lr=
|
| 218 |
" criterion = nn.CrossEntropyLoss()\n",
|
| 219 |
" \n",
|
| 220 |
" # Training loop\n",
|
|
|
|
| 214 |
" classifier.to(device)\n",
|
| 215 |
" \n",
|
| 216 |
" # Training setup\n",
|
| 217 |
+
" optimizer = torch.optim.Adam(classifier.parameters(), lr=8e-3)\n",
|
| 218 |
" criterion = nn.CrossEntropyLoss()\n",
|
| 219 |
" \n",
|
| 220 |
" # Training loop\n",
|