Karsonlin951 commited on
Commit
244d72d
1 Parent(s): f47a0a5

Training complete

Browse files
README.md ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/mt5-small
5
+ tags:
6
+ - summarization
7
+ - generated_from_trainer
8
+ metrics:
9
+ - rouge
10
+ model-index:
11
+ - name: mt5-small-finetuned-amazon-en-es
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
+ # mt5-small-finetuned-amazon-en-es
19
+
20
+ This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
21
+ It achieves the following results on the evaluation set:
22
+ - Loss: 3.7408
23
+ - Rouge1: 11.6009
24
+ - Rouge2: 2.8728
25
+ - Rougel: 11.8004
26
+ - Rougelsum: 11.9188
27
+
28
+ ## Model description
29
+
30
+ More information needed
31
+
32
+ ## Intended uses & limitations
33
+
34
+ More information needed
35
+
36
+ ## Training and evaluation data
37
+
38
+ More information needed
39
+
40
+ ## Training procedure
41
+
42
+ ### Training hyperparameters
43
+
44
+ The following hyperparameters were used during training:
45
+ - learning_rate: 5.6e-05
46
+ - train_batch_size: 8
47
+ - eval_batch_size: 8
48
+ - seed: 42
49
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
50
+ - lr_scheduler_type: linear
51
+ - num_epochs: 8
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
56
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|:-------:|:---------:|
57
+ | 12.338 | 1.0 | 300 | 5.2256 | 4.1224 | 1.2022 | 4.1645 | 4.2077 |
58
+ | 5.5954 | 2.0 | 600 | 3.9721 | 9.005 | 2.1934 | 9.133 | 9.2792 |
59
+ | 4.5559 | 3.0 | 900 | 3.8552 | 9.5807 | 2.6302 | 9.511 | 9.5471 |
60
+ | 4.1858 | 4.0 | 1200 | 3.8014 | 10.8843 | 2.9462 | 10.9298 | 11.1352 |
61
+ | 3.9963 | 5.0 | 1500 | 3.7575 | 13.0171 | 4.7986 | 13.0233 | 12.9146 |
62
+ | 3.884 | 6.0 | 1800 | 3.7491 | 11.6897 | 2.8728 | 11.8886 | 12.0365 |
63
+ | 3.8293 | 7.0 | 2100 | 3.7408 | 11.4003 | 2.8728 | 11.5425 | 11.6537 |
64
+ | 3.7902 | 8.0 | 2400 | 3.7408 | 11.6009 | 2.8728 | 11.8004 | 11.9188 |
65
+
66
+
67
+ ### Framework versions
68
+
69
+ - Transformers 4.46.2
70
+ - Pytorch 2.5.1+cu121
71
+ - Tokenizers 0.20.3
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.46.2"
6
+ }
runs/Dec04_19-56-20_2c73218c73a2/events.out.tfevents.1733343174.2c73218c73a2.632.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34a79f41d3d10df45c1faf0db58da1ba18e5edea7176400d7f1a10f151ca3c76
3
+ size 562