Update config.json
This commit is contained in:
parent
f31d29dd8d
commit
9924d1fb3f
11
config.json
11
config.json
|
@ -1,5 +1,4 @@
|
||||||
{
|
{
|
||||||
"_num_labels": 3,
|
|
||||||
"activation_dropout": 0.0,
|
"activation_dropout": 0.0,
|
||||||
"activation_function": "swish",
|
"activation_function": "swish",
|
||||||
"add_bias_logits": false,
|
"add_bias_logits": false,
|
||||||
|
@ -10,7 +9,7 @@
|
||||||
"attention_dropout": 0.0,
|
"attention_dropout": 0.0,
|
||||||
"bad_words_ids": [
|
"bad_words_ids": [
|
||||||
[
|
[
|
||||||
58378
|
65000
|
||||||
]
|
]
|
||||||
],
|
],
|
||||||
"bos_token_id": 0,
|
"bos_token_id": 0,
|
||||||
|
@ -20,13 +19,15 @@
|
||||||
"decoder_ffn_dim": 2048,
|
"decoder_ffn_dim": 2048,
|
||||||
"decoder_layerdrop": 0.0,
|
"decoder_layerdrop": 0.0,
|
||||||
"decoder_layers": 6,
|
"decoder_layers": 6,
|
||||||
"decoder_start_token_id": 58378,
|
"decoder_start_token_id": 65000,
|
||||||
"dropout": 0.1,
|
"dropout": 0.1,
|
||||||
"encoder_attention_heads": 8,
|
"encoder_attention_heads": 8,
|
||||||
"encoder_ffn_dim": 2048,
|
"encoder_ffn_dim": 2048,
|
||||||
"encoder_layerdrop": 0.0,
|
"encoder_layerdrop": 0.0,
|
||||||
"encoder_layers": 6,
|
"encoder_layers": 6,
|
||||||
"eos_token_id": 0,
|
"eos_token_id": 0,
|
||||||
|
"extra_pos_embeddings": 65001,
|
||||||
|
"force_bos_token_to_be_generated": false,
|
||||||
"id2label": {
|
"id2label": {
|
||||||
"0": "LABEL_0",
|
"0": "LABEL_0",
|
||||||
"1": "LABEL_1",
|
"1": "LABEL_1",
|
||||||
|
@ -46,8 +47,8 @@
|
||||||
"normalize_embedding": false,
|
"normalize_embedding": false,
|
||||||
"num_beams": 6,
|
"num_beams": 6,
|
||||||
"num_hidden_layers": 6,
|
"num_hidden_layers": 6,
|
||||||
"pad_token_id": 58378,
|
"pad_token_id": 65000,
|
||||||
"scale_embedding": true,
|
"scale_embedding": true,
|
||||||
"static_position_embeddings": true,
|
"static_position_embeddings": true,
|
||||||
"vocab_size": 58379
|
"vocab_size": 65001
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue