tuning finished

ai-glimpse · Feb 13, 2025 · 83c7cc4 · 83c7cc4
1 parent 2ef69d8
commit 83c7cc4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/toynlp/nnlm/train.py b/toynlp/nnlm/train.py
@@ -166,7 +166,7 @@ def run(config: NNLMConfig):
             embedding_dim=100,
             hidden_dim=60,
             with_direct_connection=False,
-            with_dropout=False,
+            with_dropout=True,
             dropout_rate=0.2,
         ),
         optimizer=OptimizerConfig(