Hugging Face's logo

zeyadusf
/
FlanT5Summarization-samsum

Summarization
PEFT
Safetensors
English
t5
Generated from Trainer
Model card Files Community
zeyadusf commited on
Commit
93cf0dd
•
1 Parent(s): f40a0e2

Model save

Browse files
Files changed (1) hide show
  1. README.md +104 -0
README.md ADDED
@@ -0,0 +1,104 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: google/flan-t5-large
3
+ datasets:
4
+ - samsum
5
+ library_name: peft
6
+ license: apache-2.0
7
+ metrics:
8
+ - rouge
9
+ tags:
10
+ - generated_from_trainer
11
+ model-index:
12
+ - name: FlanT5Summarization-samsum
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/daigt_team/Summarization%20by%20Finetuning%20FlanT5-LoRA/runs/bzfwtjcj)
20
+ # FlanT5Summarization-samsum
21
+
22
+ This model is a fine-tuned version of [google/flan-t5-large](https://huggingface.co/google/flan-t5-large) on the samsum dataset.
23
+ It achieves the following results on the evaluation set:
24
+ - Loss: 0.3001
25
+ - Rouge1: 0.2788
26
+ - Rouge2: 0.1310
27
+ - Rougel: 0.2363
28
+ - Rougelsum: 0.2369
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 3e-05
48
+ - train_batch_size: 128
49
+ - eval_batch_size: 64
50
+ - seed: 42
51
+ - gradient_accumulation_steps: 4
52
+ - total_train_batch_size: 512
53
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
+ - lr_scheduler_type: cosine
55
+ - lr_scheduler_warmup_ratio: 0.1
56
+ - num_epochs: 3
57
+
58
+ ### Training results
59
+
60
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
61
+ |:-------------:|:------:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
62
+ | 1.1072 | 0.0866 | 5 | 0.9165 | 0.2705 | 0.1135 | 0.2226 | 0.2229 |
63
+ | 1.1039 | 0.1732 | 10 | 0.9080 | 0.2709 | 0.1138 | 0.2230 | 0.2234 |
64
+ | 1.0848 | 0.2597 | 15 | 0.8917 | 0.2706 | 0.1137 | 0.2228 | 0.2231 |
65
+ | 1.0706 | 0.3463 | 20 | 0.8654 | 0.2709 | 0.1142 | 0.2232 | 0.2234 |
66
+ | 1.0461 | 0.4329 | 25 | 0.8336 | 0.2706 | 0.1140 | 0.2228 | 0.2232 |
67
+ | 1.0187 | 0.5195 | 30 | 0.7960 | 0.2718 | 0.1145 | 0.2240 | 0.2243 |
68
+ | 0.9774 | 0.6061 | 35 | 0.7532 | 0.2723 | 0.1152 | 0.2250 | 0.2253 |
69
+ | 0.9326 | 0.6926 | 40 | 0.7064 | 0.2726 | 0.1153 | 0.2253 | 0.2257 |
70
+ | 0.8834 | 0.7792 | 45 | 0.6570 | 0.2728 | 0.1160 | 0.2259 | 0.2261 |
71
+ | 0.833 | 0.8658 | 50 | 0.6080 | 0.2734 | 0.1161 | 0.2262 | 0.2263 |
72
+ | 0.7871 | 0.9524 | 55 | 0.5614 | 0.2726 | 0.1156 | 0.2260 | 0.2260 |
73
+ | 0.735 | 1.0390 | 60 | 0.5180 | 0.2731 | 0.1169 | 0.2262 | 0.2264 |
74
+ | 0.6978 | 1.1255 | 65 | 0.4802 | 0.2736 | 0.1179 | 0.2275 | 0.2276 |
75
+ | 0.6464 | 1.2121 | 70 | 0.4482 | 0.2741 | 0.1188 | 0.2283 | 0.2286 |
76
+ | 0.6175 | 1.2987 | 75 | 0.4222 | 0.2742 | 0.1193 | 0.2291 | 0.2292 |
77
+ | 0.5722 | 1.3853 | 80 | 0.4007 | 0.2740 | 0.1187 | 0.2287 | 0.2287 |
78
+ | 0.5443 | 1.4719 | 85 | 0.3834 | 0.2730 | 0.1180 | 0.2282 | 0.2282 |
79
+ | 0.5203 | 1.5584 | 90 | 0.3692 | 0.2740 | 0.1192 | 0.2293 | 0.2293 |
80
+ | 0.4851 | 1.6450 | 95 | 0.3568 | 0.2744 | 0.1201 | 0.2300 | 0.2302 |
81
+ | 0.4619 | 1.7316 | 100 | 0.3466 | 0.2746 | 0.1201 | 0.2304 | 0.2305 |
82
+ | 0.4484 | 1.8182 | 105 | 0.3379 | 0.2754 | 0.1218 | 0.2314 | 0.2319 |
83
+ | 0.4357 | 1.9048 | 110 | 0.3305 | 0.2766 | 0.1241 | 0.2325 | 0.2330 |
84
+ | 0.4246 | 1.9913 | 115 | 0.3243 | 0.2772 | 0.1254 | 0.2338 | 0.2341 |
85
+ | 0.4074 | 2.0779 | 120 | 0.3190 | 0.2776 | 0.1263 | 0.2343 | 0.2347 |
86
+ | 0.3965 | 2.1645 | 125 | 0.3144 | 0.2775 | 0.1264 | 0.2342 | 0.2345 |
87
+ | 0.3922 | 2.2511 | 130 | 0.3105 | 0.2776 | 0.1266 | 0.2344 | 0.2347 |
88
+ | 0.3861 | 2.3377 | 135 | 0.3073 | 0.2786 | 0.1289 | 0.2357 | 0.2362 |
89
+ | 0.382 | 2.4242 | 140 | 0.3048 | 0.2782 | 0.1289 | 0.2354 | 0.2358 |
90
+ | 0.3807 | 2.5108 | 145 | 0.3029 | 0.2787 | 0.1297 | 0.2359 | 0.2364 |
91
+ | 0.3717 | 2.5974 | 150 | 0.3016 | 0.2787 | 0.1303 | 0.2363 | 0.2367 |
92
+ | 0.3708 | 2.6840 | 155 | 0.3008 | 0.2788 | 0.1305 | 0.2363 | 0.2368 |
93
+ | 0.372 | 2.7706 | 160 | 0.3003 | 0.2789 | 0.1310 | 0.2365 | 0.2370 |
94
+ | 0.3696 | 2.8571 | 165 | 0.3002 | 0.2788 | 0.1310 | 0.2363 | 0.2369 |
95
+ | 0.3646 | 2.9437 | 170 | 0.3001 | 0.2788 | 0.1310 | 0.2363 | 0.2369 |
96
+
97
+
98
+ ### Framework versions
99
+
100
+ - PEFT 0.12.0
101
+ - Transformers 4.43.2
102
+ - Pytorch 2.1.2
103
+ - Datasets 2.20.0
104
+ - Tokenizers 0.19.1