diff --git a/README.md b/README.md index 7453f6f..2d12607 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@ tags: - summarization license: apache-2.0 datasets: -- cnndm +- cnn_dailymail - xsum thumbnail: https://huggingface.co/front/thumbnails/distilbart_medium.png --- @@ -24,5 +24,6 @@ This checkpoint should be loaded into `BartForConditionalGeneration.from_pretrai | bart-large-xsum (baseline) | 406 | 229 | 1 | 21.85 | 36.50 | | distilbart-xsum-12-6 | 306 | 137 | 1.68 | 22.12 | 36.99 | | bart-large-cnn (baseline) | 406 | 381 | 1 | 21.06 | 30.63 | +| distilbart-12-3-cnn | 255 | 214 | 1.78 | 20.57 | 30.00 | | 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 | 2.09 | 20.17 | 29.70 |