voidful commited on
Commit
0f7e0a1
1 Parent(s): 37c7d0b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +24 -12
config.json CHANGED
@@ -49,22 +49,34 @@
49
  "scale_embedding": false,
50
  "task_specific_params": {
51
  "summarization": {
52
- "length_penalty": 1.0,
53
- "max_length": 128,
54
- "min_length": 12,
55
- "num_beams": 4
 
 
 
 
56
  },
57
  "summarization_cnn": {
58
- "length_penalty": 2.0,
59
- "max_length": 142,
60
- "min_length": 56,
61
- "num_beams": 4
 
 
 
 
62
  },
63
  "summarization_xsum": {
64
- "length_penalty": 1.0,
65
- "max_length": 62,
66
- "min_length": 11,
67
- "num_beams": 6
 
 
 
 
68
  }
69
  },
70
  "tie_word_embeddings": false,
 
49
  "scale_embedding": false,
50
  "task_specific_params": {
51
  "summarization": {
52
+ "no_repeat_ngram_size": 3,
53
+ "do_sample": true,
54
+ "top_k": 100,
55
+ "top_p": 0.95,
56
+ "typical_p": 0.95,
57
+ "temperature": 0.80,
58
+ "pad_token_id": 0,
59
+ "max_length": 256,
60
  },
61
  "summarization_cnn": {
62
+ "no_repeat_ngram_size": 3,
63
+ "do_sample": true,
64
+ "top_k": 100,
65
+ "top_p": 0.95,
66
+ "typical_p": 0.95,
67
+ "temperature": 0.80,
68
+ "pad_token_id": 0,
69
+ "max_length": 256,
70
  },
71
  "summarization_xsum": {
72
+ "no_repeat_ngram_size": 3,
73
+ "do_sample": true,
74
+ "top_k": 100,
75
+ "top_p": 0.95,
76
+ "typical_p": 0.95,
77
+ "temperature": 0.80,
78
+ "pad_token_id": 0,
79
+ "max_length": 256,
80
  }
81
  },
82
  "tie_word_embeddings": false,