Update config.json
This commit is contained in:
parent
cc6128f93e
commit
18411abee9
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue