diff --git a/lectures/lecture-13/CBOW_Classifier.ipynb b/lectures/lecture-13/CBOW_Classifier.ipynb index 2245de2eb2eae709573943de9320fe10ae6f108f..4c05bf39da48652ecc5ad5bd34ba1b2e921844b1 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",