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"
],
"attention_dropout": 0.0,
"bad_words_ids": [
[
58378
]
],
"bos_token_id": 0,
"classif_dropout": 0.0,
"d_model": 512,
@ -34,6 +39,7 @@
"LABEL_1": 1,
"LABEL_2": 2
},
"max_length": 512,
"max_position_embeddings": 512,
"model_type": "marian",
"normalize_before": false,