Update config.json

This commit is contained in:
system 2020-05-07 02:25:31 +00:00
parent cc6128f93e
commit 18411abee9
1 changed files with 6 additions and 0 deletions

View File

@ -8,6 +8,11 @@
"MarianMTModel" "MarianMTModel"
], ],
"attention_dropout": 0.0, "attention_dropout": 0.0,
"bad_words_ids": [
[
58378
]
],
"bos_token_id": 0, "bos_token_id": 0,
"classif_dropout": 0.0, "classif_dropout": 0.0,
"d_model": 512, "d_model": 512,
@ -34,6 +39,7 @@
"LABEL_1": 1, "LABEL_1": 1,
"LABEL_2": 2 "LABEL_2": 2
}, },
"max_length": 512,
"max_position_embeddings": 512, "max_position_embeddings": 512,
"model_type": "marian", "model_type": "marian",
"normalize_before": false, "normalize_before": false,