UNIST-Eunchan
commited on
Commit
•
08748c3
1
Parent(s):
555d47d
Update README.md
Browse files
README.md
CHANGED
@@ -82,17 +82,18 @@ Paper Summarization
|
|
82 |
length_penalty=1, num_beams=2, max_length=128*4,min_length=150, no_repeat_ngram_size= 3, top_k=25,top_p=0.95)
|
83 |
|
84 |
```
|
85 |
-
- R-1 | R-2 | R-L | R-LSUM : 40.8486 | 16.3717 | 25.2937 | 33.6923 at
|
86 |
```(python)
|
87 |
model.generate(input_ids =inputs["input_ids"].to(device),
|
88 |
attention_mask=inputs["attention_mask"].to(device),
|
89 |
length_penalty=1, num_beams=1, max_length=128*2,top_p=1)
|
90 |
```
|
91 |
-
- R-1 | R-2 | R-L | R-LSUM : TBD | TBD | TBD | TBD at
|
92 |
```(python)
|
93 |
model.generate(input_ids =inputs["input_ids"].to(device),
|
94 |
attention_mask=inputs["attention_mask"].to(device),
|
95 |
-
num_beam_groups=
|
|
|
96 |
```
|
97 |
|
98 |
|
|
|
82 |
length_penalty=1, num_beams=2, max_length=128*4,min_length=150, no_repeat_ngram_size= 3, top_k=25,top_p=0.95)
|
83 |
|
84 |
```
|
85 |
+
- R-1 | R-2 | R-L | R-LSUM : 40.8486 | 16.3717 | 25.2937 | 33.6923 (refer to **PEGASUS-X's [paper](https://arxiv.org/pdf/2208.04347.pdf)**) at
|
86 |
```(python)
|
87 |
model.generate(input_ids =inputs["input_ids"].to(device),
|
88 |
attention_mask=inputs["attention_mask"].to(device),
|
89 |
length_penalty=1, num_beams=1, max_length=128*2,top_p=1)
|
90 |
```
|
91 |
+
- R-1 | R-2 | R-L | R-LSUM : TBD | TBD | TBD | TBD (**Diverse Beam-Search Decoding**) at
|
92 |
```(python)
|
93 |
model.generate(input_ids =inputs["input_ids"].to(device),
|
94 |
attention_mask=inputs["attention_mask"].to(device),
|
95 |
+
num_beam_groups=5,diversity_penalty=1.0,num_beams=5,min_length=100,max_length=128*4)
|
96 |
+
|
97 |
```
|
98 |
|
99 |
|