Hugging Face's logo

gsvann
/
mt5-small-finetuned-amazon-en-de

Text2Text Generation
Transformers
PyTorch
mt5
Generated from Trainer
Inference Endpoints
Model card Files Community
gsvann commited on
Commit
70aa57d
·
1 Parent(s): b4815db

gsvann/mt5-small-finetuned-amazon-toys-en-de

Browse files
Files changed (3) hide show
  1. README.md +14 -14
  2. pytorch_model.bin +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -17,11 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 2.6132
21
- - Rouge1: 18.5886
22
- - Rouge2: 10.4761
23
- - Rougel: 17.9705
24
- - Rougelsum: 18.2467
25
 
26
  ## Model description
27
 
@@ -40,9 +40,9 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - learning_rate: 5.6e-05
44
- - train_batch_size: 16
45
- - eval_batch_size: 16
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
@@ -52,12 +52,12 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
55
- | 3.0822 | 1.0 | 651 | 2.6232 | 17.8834 | 11.2967 | 17.4489 | 17.73 |
56
- | 2.9301 | 2.0 | 1302 | 2.6265 | 18.3784 | 11.1587 | 18.0152 | 18.1896 |
57
- | 2.8552 | 3.0 | 1953 | 2.6223 | 18.7889 | 11.1004 | 18.2222 | 18.4583 |
58
- | 2.8253 | 4.0 | 2604 | 2.6080 | 18.2031 | 10.2714 | 17.748 | 17.8395 |
59
- | 2.8044 | 5.0 | 3255 | 2.6154 | 18.5675 | 10.5794 | 17.967 | 18.2698 |
60
- | 2.7928 | 6.0 | 3906 | 2.6132 | 18.5886 | 10.4761 | 17.9705 | 18.2467 |
61
 
62
 
63
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 2.5965
21
+ - Rouge1: 19.1764
22
+ - Rouge2: 10.6855
23
+ - Rougel: 18.7602
24
+ - Rougelsum: 18.8956
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 5.5e-05
44
+ - train_batch_size: 8
45
+ - eval_batch_size: 8
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
55
+ | 2.9041 | 1.0 | 1301 | 2.6000 | 17.3749 | 10.0728 | 16.9903 | 17.0336 |
56
+ | 2.744 | 2.0 | 2602 | 2.5874 | 17.7266 | 9.2481 | 17.2785 | 17.3827 |
57
+ | 2.6641 | 3.0 | 3903 | 2.6001 | 19.0052 | 10.6312 | 18.7604 | 18.754 |
58
+ | 2.6189 | 4.0 | 5204 | 2.6012 | 18.834 | 10.1299 | 18.4209 | 18.5351 |
59
+ | 2.6029 | 5.0 | 6505 | 2.5944 | 19.3375 | 10.537 | 18.8614 | 19.0826 |
60
+ | 2.6086 | 6.0 | 7806 | 2.5965 | 19.1764 | 10.6855 | 18.7602 | 18.8956 |
61
 
62
 
63
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc66756f726ca2bb65ee3f851abac731eb68f8e31657faf0c0ce42470aa45354
3
  size 1200773058
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b50776badb8287353e3cd15c0fbdd307c70b97eba1df4f71840d2dd841f49a04
3
  size 1200773058
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0c3bf9535f955a89e82e4cc1478fe509fc4302a1aa4ff898439dda5668decf5e
3
  size 4664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49de751304fd243924c782ec9c3d405a206b4d21a73e7a1c31f29b5e28d60210
3
  size 4664