sumedh commited on
Commit
7ed11eb
1 Parent(s): e39d7fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md CHANGED
@@ -1,3 +1,23 @@
1
  ---
 
 
 
2
  license: apache-2.0
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language: en
3
+ tags:
4
+ - summarization
5
  license: apache-2.0
6
+ datasets:
7
+ - amazon_reviews_multi
8
+ thumbnail: https://huggingface.co/front/thumbnails/distilbart_medium.png
9
+
10
  ---
11
+ ### Usage
12
+
13
+ This checkpoint should be loaded into `BartForConditionalGeneration.from_pretrained`. See the [BART docs](https://huggingface.co/transformers/model_doc/bart.html?#transformers.BartForConditionalGeneration) for more information.
14
+
15
+ ---
16
+
17
+ ### Metrics for DistilBART models
18
+
19
+ | Model Name | MM Params | Inference Time (MS) | Speedup | Rouge 1 | Rouge-L |
20
+ |:---------------------------------|------------:|----------------------:|----------:|----------:|----------:|
21
+ | distilbart-cnn-12-6-amazonreviews| 306 | 137 | 1.68 | 10.557| 33.37 |
22
+
23
+ ---