Hugging Face's logo

theojolliffe
/
distilbart-cnn-arxiv-pubmed-pubmed-earlystopping

Text2Text Generation
Transformers
PyTorch
TensorBoard
bart
generated_from_trainer
Inference Endpoints
Model card Files Metrics Community
theojolliffe commited on
Commit
f6f8f09
•
1 Parent(s): 6eb965d

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +78 -0
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - rouge
7
+ model-index:
8
+ - name: distilbart-cnn-arxiv-pubmed-pubmed-earlystopping
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # distilbart-cnn-arxiv-pubmed-pubmed-earlystopping
16
+
17
+ This model is a fine-tuned version of [theojolliffe/distilbart-cnn-arxiv-pubmed-pubmed](https://huggingface.co/theojolliffe/distilbart-cnn-arxiv-pubmed-pubmed) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.8596
20
+ - Rouge1: 53.4491
21
+ - Rouge2: 35.0041
22
+ - Rougel: 37.2742
23
+ - Rougelsum: 50.9867
24
+ - Gen Len: 142.0
25
+
26
+ ## Model description
27
+
28
+ More information needed
29
+
30
+ ## Intended uses & limitations
31
+
32
+ More information needed
33
+
34
+ ## Training and evaluation data
35
+
36
+ More information needed
37
+
38
+ ## Training procedure
39
+
40
+ ### Training hyperparameters
41
+
42
+ The following hyperparameters were used during training:
43
+ - learning_rate: 2e-05
44
+ - train_batch_size: 2
45
+ - eval_batch_size: 2
46
+ - seed: 42
47
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
+ - lr_scheduler_type: linear
49
+ - num_epochs: 20
50
+ - mixed_precision_training: Native AMP
51
+
52
+ ### Training results
53
+
54
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:--------:|
56
+ | No log | 0.31 | 125 | 1.3772 | 50.6084 | 30.8075 | 32.6113 | 47.883 | 142.0 |
57
+ | No log | 0.63 | 250 | 1.2423 | 52.1758 | 31.6326 | 32.9448 | 49.8089 | 141.6296 |
58
+ | No log | 0.94 | 375 | 1.1223 | 52.3494 | 32.3508 | 35.3638 | 49.6019 | 142.0 |
59
+ | 1.3557 | 1.26 | 500 | 1.1004 | 51.8935 | 32.8506 | 35.521 | 49.6249 | 142.0 |
60
+ | 1.3557 | 1.57 | 625 | 1.0600 | 50.8085 | 31.0397 | 34.2021 | 48.2264 | 141.5741 |
61
+ | 1.3557 | 1.88 | 750 | 0.9834 | 53.0701 | 34.0699 | 36.4029 | 51.043 | 142.0 |
62
+ | 1.3557 | 2.2 | 875 | 0.9554 | 53.4385 | 34.2976 | 36.8142 | 51.1262 | 141.9444 |
63
+ | 0.868 | 2.51 | 1000 | 0.9256 | 52.2123 | 32.7568 | 34.5883 | 49.8566 | 142.0 |
64
+ | 0.868 | 2.83 | 1125 | 0.8944 | 53.8062 | 34.6687 | 36.9645 | 51.5162 | 142.0 |
65
+ | 0.868 | 3.14 | 1250 | 0.9290 | 53.1356 | 34.1301 | 37.7713 | 50.762 | 141.9074 |
66
+ | 0.868 | 3.45 | 1375 | 0.9017 | 53.4455 | 35.0572 | 37.3033 | 50.9773 | 142.0 |
67
+ | 0.6252 | 3.77 | 1500 | 0.8519 | 53.9228 | 35.5575 | 38.9119 | 51.5202 | 142.0 |
68
+ | 0.6252 | 4.08 | 1625 | 0.8991 | 54.4223 | 36.3072 | 38.5771 | 51.9874 | 141.9074 |
69
+ | 0.6252 | 4.4 | 1750 | 0.8857 | 53.4105 | 35.348 | 37.5814 | 50.8842 | 142.0 |
70
+ | 0.6252 | 4.71 | 1875 | 0.8596 | 53.4491 | 35.0041 | 37.2742 | 50.9867 | 142.0 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.18.0
76
+ - Pytorch 1.11.0+cu113
77
+ - Datasets 2.2.1
78
+ - Tokenizers 0.12.1