facebook
/
bart-large
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.json

This commit is contained in:
system 2020-10-26 21:24:10 +00:00
parent 6000698fa9
commit f8275ef803
1 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
config.json
View File

@ -1,16 +1,16 @@
{ {
"_num_labels": 3, "activation_dropout": 0.1,
"activation_dropout": 0.0,
"activation_function": "gelu", "activation_function": "gelu",
"add_bias_logits": false,
"add_final_layer_norm": false, "add_final_layer_norm": false,
"architectures": [ "architectures": [
"BartModel", "BartModel",
"BartForMaskedLM", "BartForConditionalGeneration",
"BartForSequenceClassification" "BartForSequenceClassification"
], ],
"attention_dropout": 0.0, "attention_dropout": 0.1,
"bos_token_id": 0, "bos_token_id": 0,
"classif_dropout": 0.0, "classif_dropout": 0.1,
"d_model": 1024, "d_model": 1024,
"decoder_attention_heads": 16, "decoder_attention_heads": 16,
"decoder_ffn_dim": 4096, "decoder_ffn_dim": 4096,
@ -18,6 +18,7 @@
"decoder_layers": 12, "decoder_layers": 12,
"decoder_start_token_id": 2, "decoder_start_token_id": 2,
"dropout": 0.1, "dropout": 0.1,
"early_stopping": true,
"encoder_attention_heads": 16, "encoder_attention_heads": 16,
"encoder_ffn_dim": 4096, "encoder_ffn_dim": 4096,
"encoder_layerdrop": 0.0, "encoder_layerdrop": 0.0,
@ -39,18 +40,28 @@
"model_type": "bart", "model_type": "bart",
"normalize_before": false, "normalize_before": false,
"num_hidden_layers": 12, "num_hidden_layers": 12,
"output_past": false,
"pad_token_id": 1, "pad_token_id": 1,
"prefix": " ",
"scale_embedding": false, "scale_embedding": false,
"num_beams": 4,
"no_repeat_ngram_size": 3,
"task_specific_params": { "task_specific_params": {
"summarization": { "summarization": {
"early_stopping": true, "length_penalty": 1.0,
"max_length": 128,
"min_length": 12,
"num_beams": 4
},
"summarization_cnn": {
"length_penalty": 2.0, "length_penalty": 2.0,
"max_length": 142, "max_length": 142,
"min_length": 56, "min_length": 56,
"no_repeat_ngram_size": 3,
"num_beams": 4 "num_beams": 4
},
"summarization_xsum": {
"length_penalty": 1.0,
"max_length": 62,
"min_length": 11,
"num_beams": 6
} }
}, },
"vocab_size": 50265 "vocab_size": 50265