Update README.md

This commit is contained in:
system 2020-07-01 15:46:23 +00:00
parent 0cf8c9af15
commit 0f113b672c
1 changed files with 11 additions and 4 deletions

View File

@ -1,3 +1,14 @@
---
language: english
tags:
- summarization
license: apache-2.0
datasets:
- cnndm
- xsum
- thumbnail: https://huggingface.co/front/thumbnails/distilbart_medium.png
---
### Usage ### Usage
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. 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.
@ -15,7 +26,3 @@ This checkpoint should be loaded into `BartForConditionalGeneration.from_pretrai
| bart-large-cnn (baseline) | 406 | 381 | 1 | 21.06 | 30.63 | | bart-large-cnn (baseline) | 406 | 381 | 1 | 21.06 | 30.63 |
| distilbart-12-6-cnn | 306 | 307 | 1.24 | 21.26 | 30.59 | | distilbart-12-6-cnn | 306 | 307 | 1.24 | 21.26 | 30.59 |
| distilbart-6-6-cnn | 230 | 182 | 1.69 | 20.17 | 29.70 | | distilbart-6-6-cnn | 230 | 182 | 1.69 | 20.17 | 29.70 |
-thumbnail: https://huggingface.co/front/thumbnails/distilbart_medium.png