SurAyush commited on
Commit
74a4bd4
·
verified ·
1 Parent(s): 7da9d8a

Training Complete

Browse files
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google-t5/t5-small
5
+ tags:
6
+ - summarization
7
+ - generated_from_trainer
8
+ metrics:
9
+ - rouge
10
+ model-index:
11
+ - name: news-summarizer-t5
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ # news-summarizer-t5
19
+
20
+ This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
21
+ It achieves the following results on the evaluation set:
22
+ - Loss: 0.6177
23
+ - Model Preparation Time: 0.0049
24
+ - Rouge1: 19.8849
25
+ - Rouge2: 17.9939
26
+ - Rougel: 19.5328
27
+ - Rougelsum: 19.5918
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 0.0005
47
+ - train_batch_size: 8
48
+ - eval_batch_size: 8
49
+ - seed: 42
50
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
51
+ - lr_scheduler_type: linear
52
+ - num_epochs: 8
53
+ - mixed_precision_training: Native AMP
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | Model Preparation Time | Rouge1 | Rouge2 | Rougel | Rougelsum |
58
+ |:-------------:|:-----:|:----:|:---------------:|:----------------------:|:-------:|:-------:|:-------:|:---------:|
59
+ | 0.8689 | 1.0 | 251 | 0.6581 | 0.0049 | 18.8745 | 16.2314 | 18.1991 | 18.3287 |
60
+ | 0.6629 | 2.0 | 502 | 0.6385 | 0.0049 | 19.3705 | 17.1277 | 18.8685 | 18.9594 |
61
+ | 0.6114 | 3.0 | 753 | 0.6294 | 0.0049 | 19.3951 | 17.2113 | 18.9315 | 18.9848 |
62
+ | 0.571 | 4.0 | 1004 | 0.6197 | 0.0049 | 19.8684 | 17.8234 | 19.4646 | 19.5401 |
63
+ | 0.5451 | 5.0 | 1255 | 0.6193 | 0.0049 | 19.8981 | 17.9851 | 19.5083 | 19.5177 |
64
+ | 0.5194 | 6.0 | 1506 | 0.6203 | 0.0049 | 19.8675 | 17.9521 | 19.5434 | 19.6046 |
65
+ | 0.4894 | 7.0 | 1757 | 0.6166 | 0.0049 | 19.8622 | 17.9616 | 19.4791 | 19.5669 |
66
+ | 0.4872 | 8.0 | 2008 | 0.6177 | 0.0049 | 19.8849 | 17.9939 | 19.5328 | 19.5918 |
67
+
68
+
69
+ ### Framework versions
70
+
71
+ - Transformers 4.50.0
72
+ - Pytorch 2.6.0+cu124
73
+ - Datasets 3.5.0
74
+ - Tokenizers 0.21.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.50.0"
6
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1978d63076add613baa2e747fb33617de48c19b1d8646188129ef681914a3038
3
  size 242041896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1814bd46c902759a4e1e424aab2eb3bcdc9bbd25550a754c8c304394b811e543
3
  size 242041896
runs/Mar31_15-36-39_8e470de03c02/events.out.tfevents.1743435457.8e470de03c02.749.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c07eb974cb3c368ccc7a18b35a4801087903eefe6f110de13e2d560b2eff38a1
3
- size 11987
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15f03c2978ba34f9f8ae32b7696cc55f7869d881e0e7bd0ab23606fbe3c52396
3
+ size 12881