Changed num_beams to 4

2021-01-18 08:58:49 +01:00 · 2021-01-18 08:58:49 +01:00 · 2ae01f3349
parent 9bf1bf0262
commit 2ae01f3349
1 changed files with 2 additions and 2 deletions
--- a/config.json
+++ b/config.json
@ -44,10 +44,10 @@
  "model_type": "marian",
  "normalize_before": false,
  "normalize_embedding": false,
-  "num_beams": 6,
+  "num_beams": 4,
  "num_hidden_layers": 6,
  "pad_token_id": 58100,
  "scale_embedding": true,
  "static_position_embeddings": true,
  "vocab_size": 58101
-}
+}