Update config.json with task-specific parameters
Browse files- config.json +11 -1
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "t5-small",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
@@ -26,6 +26,16 @@
|
|
26 |
"relative_attention_max_distance": 128,
|
27 |
"relative_attention_num_buckets": 32,
|
28 |
"task_specific_params": {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
"summarization": {
|
30 |
"early_stopping": true,
|
31 |
"length_penalty": 2.0,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "willwade/t5-small-spoken-typo",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
|
|
26 |
"relative_attention_max_distance": 128,
|
27 |
"relative_attention_num_buckets": 32,
|
28 |
"task_specific_params": {
|
29 |
+
"text_correction": {
|
30 |
+
"max_length": 64,
|
31 |
+
"num_beams": 4,
|
32 |
+
"early_stopping": true,
|
33 |
+
"length_penalty": 1.0,
|
34 |
+
"no_repeat_ngram_size": 2,
|
35 |
+
"temperature": 1.0,
|
36 |
+
"top_k": 50,
|
37 |
+
"top_p": 0.95
|
38 |
+
},
|
39 |
"summarization": {
|
40 |
"early_stopping": true,
|
41 |
"length_penalty": 2.0,
|