Update config.json
This commit is contained in:
parent
6fd4954682
commit
d23db9512f
@ -12,7 +12,7 @@
|
|||||||
"decoder_layers": 12,
|
"decoder_layers": 12,
|
||||||
"do_sample": false,
|
"do_sample": false,
|
||||||
"dropout": 0.1,
|
"dropout": 0.1,
|
||||||
"early_stopping": false,
|
"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,
|
||||||
@ -35,9 +35,9 @@
|
|||||||
"LABEL_2": 2
|
"LABEL_2": 2
|
||||||
},
|
},
|
||||||
"length_penalty": 2.0,
|
"length_penalty": 2.0,
|
||||||
"max_length": 20,
|
"max_length": 142,
|
||||||
"max_position_embeddings": 1024,
|
"max_position_embeddings": 1024,
|
||||||
"min_length": 0,
|
"min_length": 56,
|
||||||
"model_type": "bart",
|
"model_type": "bart",
|
||||||
"no_repeat_ngram_size": 3,
|
"no_repeat_ngram_size": 3,
|
||||||
"num_beams": 4,
|
"num_beams": 4,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user