siddheshtv commited on
Commit
26c5d25
·
verified ·
1 Parent(s): 6c1d524

End of training

Browse files
README.md ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google-t5/t5-small
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - rouge
8
+ model-index:
9
+ - name: abstractive_summarization
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
+ # abstractive_summarization
17
+
18
+ This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 2.0699
21
+ - Rouge1: 0.166
22
+ - Rouge2: 0.1297
23
+ - Rougel: 0.1594
24
+ - Rougelsum: 0.1593
25
+ - Gen Len: 18.9974
26
+
27
+ ## Model description
28
+
29
+ More information needed
30
+
31
+ ## Intended uses & limitations
32
+
33
+ More information needed
34
+
35
+ ## Training and evaluation data
36
+
37
+ More information needed
38
+
39
+ ## Training procedure
40
+
41
+ ### Training hyperparameters
42
+
43
+ The following hyperparameters were used during training:
44
+ - learning_rate: 2e-05
45
+ - train_batch_size: 4
46
+ - eval_batch_size: 4
47
+ - seed: 42
48
+ - gradient_accumulation_steps: 2
49
+ - total_train_batch_size: 8
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
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 | Gen Len |
58
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
59
+ | 2.465 | 1.0 | 1658 | 2.1944 | 0.1613 | 0.1244 | 0.1538 | 0.1537 | 18.996 |
60
+ | 2.3525 | 2.0 | 3316 | 2.1101 | 0.1646 | 0.128 | 0.1572 | 0.1571 | 18.9974 |
61
+ | 2.2844 | 3.0 | 4974 | 2.0779 | 0.1655 | 0.1291 | 0.1587 | 0.1586 | 18.9965 |
62
+ | 2.2874 | 4.0 | 6632 | 2.0699 | 0.166 | 0.1297 | 0.1594 | 0.1593 | 18.9974 |
63
+
64
+
65
+ ### Framework versions
66
+
67
+ - Transformers 4.41.2
68
+ - Pytorch 2.3.0+cu121
69
+ - Datasets 2.20.0
70
+ - Tokenizers 0.19.1
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "decoder_start_token_id": 0,
3
+ "eos_token_id": 1,
4
+ "pad_token_id": 0,
5
+ "transformers_version": "4.41.2"
6
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d206d69696831d4ba098fed73937433a83025dc50670a78444d0ad5f29d41688
3
  size 242041896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b2809c6f438a86344c8b363b712cbdb9089d9969aef92ce4d9473d28635d2b8
3
  size 242041896
runs/Jul14_13-23-36_5b125a660546/events.out.tfevents.1720963420.5b125a660546.3465.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ccb6c6c63cc8f7e0b6a03afbc31c022985134ee64745035316ba51988ed2c227
3
- size 10143
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90511b0c2cdbc2f80b47edca753f6f398bb44368241c30dc89237b8722de51b0
3
+ size 11022