Heinrich Dinkel commited on
Commit
5340ead
·
1 Parent(s): 0f35cd3
Files changed (1) hide show
  1. 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=1e-3)\n",
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",