From a572b5ef74e58a3a9353c07c5e41b64cb773f5b2 Mon Sep 17 00:00:00 2001
From: Marco Kuhlmann <marco.kuhlmann@liu.se>
Date: Sun, 24 Jan 2021 20:35:43 +0100
Subject: [PATCH] Fix a hyperparameter
---
lectures/lecture-13/CBOW_Classifier.ipynb | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/lectures/lecture-13/CBOW_Classifier.ipynb b/lectures/lecture-13/CBOW_Classifier.ipynb
index 2245de2..4c05bf3 100644
--- a/lectures/lecture-13/CBOW_Classifier.ipynb
+++ b/lectures/lecture-13/CBOW_Classifier.ipynb
@@ -306,7 +306,7 @@
"\n",
"def train(n_epochs=10, batch_size=24, lr=1e-2, stop_early=False):\n",
" # Initialize the model\n",
- " model = CBOW(len(vocab), 50, 5)\n",
+ " model = CBOW(len(vocab), 30, 5)\n",
"\n",
" # Initialize the optimizer\n",
" optimizer = optim.Adam(model.parameters(), lr=lr)\n",
--
GitLab