José Ángel González
commited on
Commit
·
7c71bf4
1
Parent(s):
fcd619f
Update config.json
Browse filesGenerate parameters in config
- config.json +7 -0
config.json
CHANGED
|
@@ -44,6 +44,13 @@
|
|
| 44 |
"num_beams": 4
|
| 45 |
}
|
| 46 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
"mask_sent_token_id": 6,
|
| 48 |
"mask_token_id": 5,
|
| 49 |
"max_position_embeddings": 512,
|
|
|
|
| 44 |
"num_beams": 4
|
| 45 |
}
|
| 46 |
},
|
| 47 |
+
"early_stopping": true,
|
| 48 |
+
"repetition_penalty": 2.0,
|
| 49 |
+
"max_length": 150,
|
| 50 |
+
"min_length": 25,
|
| 51 |
+
"no_repeat_ngram_size": 3,
|
| 52 |
+
"num_beams": 4,
|
| 53 |
+
|
| 54 |
"mask_sent_token_id": 6,
|
| 55 |
"mask_token_id": 5,
|
| 56 |
"max_position_embeddings": 512,
|