Changed num_beams to 4

This commit is contained in:
Patrick von Platen 2021-01-18 08:58:49 +01:00
parent 9bf1bf0262
commit 2ae01f3349
1 changed files with 2 additions and 2 deletions

View File

@ -44,10 +44,10 @@
"model_type": "marian", "model_type": "marian",
"normalize_before": false, "normalize_before": false,
"normalize_embedding": false, "normalize_embedding": false,
"num_beams": 6, "num_beams": 4,
"num_hidden_layers": 6, "num_hidden_layers": 6,
"pad_token_id": 58100, "pad_token_id": 58100,
"scale_embedding": true, "scale_embedding": true,
"static_position_embeddings": true, "static_position_embeddings": true,
"vocab_size": 58101 "vocab_size": 58101
} }