scott156 commited on
Commit
8d47ca2
1 Parent(s): 2c39299

End of training

Browse files
Files changed (3) hide show
  1. README.md +70 -0
  2. generation_config.json +10 -0
  3. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: allenai/led-base-16384
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - rouge
8
+ model-index:
9
+ - name: LED-Base-NSPCC
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # LED-Base-NSPCC
17
+
18
+ This model is a fine-tuned version of [allenai/led-base-16384](https://huggingface.co/allenai/led-base-16384) on an unknown dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 1.8734
21
+ - Rouge1: 0.4910
22
+ - Rouge2: 0.2207
23
+ - Rougel: 0.2847
24
+ - Rougelsum: 0.2840
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: 0.0003
44
+ - train_batch_size: 4
45
+ - eval_batch_size: 4
46
+ - seed: 42
47
+ - gradient_accumulation_steps: 4
48
+ - total_train_batch_size: 16
49
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
+ - lr_scheduler_type: cosine
51
+ - lr_scheduler_warmup_ratio: 0.03
52
+ - num_epochs: 4
53
+ - mixed_precision_training: Native AMP
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
58
+ |:-------------:|:------:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
59
+ | 2.4662 | 0.9947 | 47 | 1.9451 | 0.4528 | 0.1809 | 0.2560 | 0.2558 |
60
+ | 1.6508 | 1.9894 | 94 | 1.8497 | 0.4889 | 0.2146 | 0.2720 | 0.2716 |
61
+ | 1.2549 | 2.9841 | 141 | 1.8268 | 0.4812 | 0.2092 | 0.2756 | 0.2753 |
62
+ | 0.9955 | 3.9788 | 188 | 1.8734 | 0.4910 | 0.2207 | 0.2847 | 0.2840 |
63
+
64
+
65
+ ### Framework versions
66
+
67
+ - Transformers 4.40.1
68
+ - Pytorch 2.2.1+cu121
69
+ - Datasets 2.19.0
70
+ - Tokenizers 0.19.1
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 2,
4
+ "eos_token_id": 2,
5
+ "max_new_tokens": 400,
6
+ "no_repeat_ngram_size": 5,
7
+ "num_beams": 3,
8
+ "pad_token_id": 1,
9
+ "transformers_version": "4.40.1"
10
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6409416b7ea5da045c0bcac299902a848576280d4efc65e95e311a28cb332914
3
  size 647614116
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37a4fa6368af37ac1d4901afef34ac68be646f8be859eed151b23b610206c353
3
  size 647614116