diff --git a/config.json b/config.json index 18e15e0..37b41bc 100644 --- a/config.json +++ b/config.json @@ -12,7 +12,7 @@ "decoder_layers": 12, "do_sample": false, "dropout": 0.1, - "early_stopping": false, + "early_stopping": true, "encoder_attention_heads": 16, "encoder_ffn_dim": 4096, "encoder_layerdrop": 0.0, @@ -35,9 +35,9 @@ "LABEL_2": 2 }, "length_penalty": 2.0, - "max_length": 20, + "max_length": 142, "max_position_embeddings": 1024, - "min_length": 0, + "min_length": 56, "model_type": "bart", "no_repeat_ngram_size": 3, "num_beams": 4,