Hugging Face's logo

huggingface-course
/
mt5-small-finetuned-amazon-en-es

Summarization
Transformers
PyTorch
TensorFlow
TensorBoard
mt5
text2text-generation
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
2
lewtun HF staff commited on
Commit
9f9c5aa
•
1 Parent(s): 6eb9279

Training complete

Browse files
Files changed (4) hide show
  1. README.md +16 -14
  2. pytorch_model.bin +1 -1
  3. tokenizer.json +0 -0
  4. training_args.bin +1 -1
README.md CHANGED
@@ -2,7 +2,6 @@
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
- - pipeline:summarization
6
  datasets:
7
  - null
8
  metrics:
@@ -16,7 +15,7 @@ model-index:
16
  metrics:
17
  - name: Rouge1
18
  type: rouge
19
- value: 8.8272
20
  ---
21
 
22
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -26,12 +25,12 @@ should probably proofread and complete it, then remove this comment. -->
26
 
27
  This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on the None dataset.
28
  It achieves the following results on the evaluation set:
29
- - Loss: 3.3342
30
- - Rouge1: 8.8272
31
- - Rouge2: 2.5114
32
- - Rougel: 8.6749
33
- - Rougelsum: 8.6722
34
- - Gen Len: 4.2877
35
 
36
  ## Model description
37
 
@@ -56,15 +55,18 @@ The following hyperparameters were used during training:
56
  - seed: 42
57
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
58
  - lr_scheduler_type: linear
59
- - num_epochs: 3
60
 
61
  ### Training results
62
 
63
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
64
- |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
65
- | 9.4562 | 1.0 | 2202 | 3.5591 | 6.6009 | 1.7239 | 6.5036 | 6.5228 | 3.4434 |
66
- | 4.6481 | 2.0 | 4404 | 3.3600 | 7.3535 | 1.9174 | 7.2846 | 7.3053 | 3.809 |
67
- | 4.3333 | 3.0 | 6606 | 3.3342 | 8.8272 | 2.5114 | 8.6749 | 8.6722 | 4.2877 |
 
 
 
68
 
69
 
70
  ### Framework versions
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
5
  datasets:
6
  - null
7
  metrics:
 
15
  metrics:
16
  - name: Rouge1
17
  type: rouge
18
+ value: 10.8752
19
  ---
20
 
21
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
25
 
26
  This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on the None dataset.
27
  It achieves the following results on the evaluation set:
28
+ - Loss: 3.1491
29
+ - Rouge1: 10.8752
30
+ - Rouge2: 3.8695
31
+ - Rougel: 10.6991
32
+ - Rougelsum: 10.6616
33
+ - Gen Len: 5.6085
34
 
35
  ## Model description
36
 
 
55
  - seed: 42
56
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
57
  - lr_scheduler_type: linear
58
+ - num_epochs: 6
59
 
60
  ### Training results
61
 
62
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
63
+ |:-------------:|:-----:|:-----:|:---------------:|:-------:|:------:|:-------:|:---------:|:-------:|
64
+ | 9.1733 | 1.0 | 2202 | 3.4863 | 6.3629 | 1.4637 | 6.2501 | 6.2752 | 3.3302 |
65
+ | 4.4547 | 2.0 | 4404 | 3.2809 | 9.1283 | 2.992 | 8.9851 | 9.0487 | 4.7642 |
66
+ | 4.0581 | 3.0 | 6606 | 3.2108 | 10.5207 | 3.7411 | 10.2595 | 10.234 | 5.3208 |
67
+ | 3.8821 | 4.0 | 8808 | 3.1701 | 10.8636 | 4.0944 | 10.6462 | 10.6468 | 5.2453 |
68
+ | 3.7857 | 5.0 | 11010 | 3.1600 | 10.9456 | 4.5187 | 10.784 | 10.7542 | 5.691 |
69
+ | 3.7273 | 6.0 | 13212 | 3.1491 | 10.8752 | 3.8695 | 10.6991 | 10.6616 | 5.6085 |
70
 
71
 
72
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4a89d30bbdfb33b505516cfb5f81c0eca7d5521acbf7ecc4275d8cd4dc0d1229
3
  size 1200770885
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fc1cebaa89d1bddeb41b5c3a2014bfc2d70cde4af9bb38ac8c5e578f61bcf22
3
  size 1200770885
tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:528e1467127904051d0c15f6a2903cb85414605e00a9e072e543cc8884e47188
3
  size 2735
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6c696d829fb994c3c16fe17700311c5b844cb2570a26fe27827c9598487fe10
3
  size 2735