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