tomvoelker commited on
Commit
c42eb91
·
verified ·
1 Parent(s): 4425b82

Model save

Browse files
Files changed (2) hide show
  1. README.md +77 -0
  2. generation_config.json +13 -0
README.md ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - rouge
7
+ model-index:
8
+ - name: gpt22gpt2-gpt2-large-cnn-dailymail-seed42
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
+ # gpt22gpt2-gpt2-large-cnn-dailymail-seed42
16
+
17
+ This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 1.6842
20
+ - Rouge1: 0.3500
21
+ - Rouge2: 0.1477
22
+ - Rougel: 0.2169
23
+ - Rougelsum: 0.3306
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 5e-05
43
+ - train_batch_size: 8
44
+ - eval_batch_size: 16
45
+ - seed: 42
46
+ - gradient_accumulation_steps: 4
47
+ - total_train_batch_size: 32
48
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
+ - lr_scheduler_type: linear
50
+ - lr_scheduler_warmup_steps: 1000
51
+ - num_epochs: 3.0
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
56
+ |:-------------:|:------:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
57
+ | 2.6006 | 0.2229 | 2000 | 2.3891 | 0.1904 | 0.0449 | 0.1271 | 0.1782 |
58
+ | 2.1985 | 0.4458 | 4000 | 2.0162 | 0.2608 | 0.0874 | 0.1661 | 0.2452 |
59
+ | 1.9813 | 0.6687 | 6000 | 1.8433 | 0.2390 | 0.0798 | 0.1570 | 0.2267 |
60
+ | 1.8954 | 0.8916 | 8000 | 1.7652 | 0.2694 | 0.0968 | 0.1718 | 0.2533 |
61
+ | 1.5988 | 1.1145 | 10000 | 1.7400 | 0.3181 | 0.1275 | 0.1992 | 0.2998 |
62
+ | 1.5897 | 1.3374 | 12000 | 1.7119 | 0.3292 | 0.1351 | 0.2049 | 0.3107 |
63
+ | 1.5809 | 1.5603 | 14000 | 1.6926 | 0.3452 | 0.1451 | 0.2142 | 0.3262 |
64
+ | 1.575 | 1.7832 | 16000 | 1.6679 | 0.3440 | 0.1452 | 0.2149 | 0.3256 |
65
+ | 1.5302 | 2.0061 | 18000 | 1.6870 | 0.3512 | 0.1486 | 0.2168 | 0.3316 |
66
+ | 1.2726 | 2.2290 | 20000 | 1.7002 | 0.3484 | 0.1460 | 0.2149 | 0.3289 |
67
+ | 1.266 | 2.4519 | 22000 | 1.6969 | 0.3473 | 0.1461 | 0.2154 | 0.3279 |
68
+ | 1.2566 | 2.6748 | 24000 | 1.6878 | 0.3487 | 0.1469 | 0.2160 | 0.3296 |
69
+ | 1.2572 | 2.8977 | 26000 | 1.6842 | 0.3500 | 0.1477 | 0.2169 | 0.3306 |
70
+
71
+
72
+ ### Framework versions
73
+
74
+ - Transformers 4.44.2
75
+ - Pytorch 2.4.0
76
+ - Datasets 2.21.0
77
+ - Tokenizers 0.19.1
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 50256,
3
+ "decoder_start_token_id": 50256,
4
+ "early_stopping": true,
5
+ "eos_token_id": 50256,
6
+ "length_penalty": 2.0,
7
+ "max_length": 142,
8
+ "min_length": 56,
9
+ "no_repeat_ngram_size": 3,
10
+ "num_beams": 4,
11
+ "pad_token_id": 50256,
12
+ "transformers_version": "4.44.2"
13
+ }