UNIST-Eunchan commited on
Commit
555d47d
1 Parent(s): dadcbe0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -88,7 +88,14 @@ Paper Summarization
88
  attention_mask=inputs["attention_mask"].to(device),
89
  length_penalty=1, num_beams=1, max_length=128*2,top_p=1)
90
  ```
91
-
 
 
 
 
 
 
 
92
 
93
  ## Training and evaluation data
94
 
 
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=16,num_beams=16,min_length=100,max_length=128*4)
96
+ ```
97
+
98
+
99
 
100
  ## Training and evaluation data
101