Hugging Face's logo

drewcurran
/
translation_model

Text2Text Generation
Transformers
TensorBoard
Safetensors
t5
Generated from Trainer
Inference Endpoints
text-generation-inference
Model card Files Metrics Community
drewcurran commited on
Commit
4864589
•
1 Parent(s): 3946e86

End of training

Browse files
Files changed (3) hide show
  1. README.md +20 -12
  2. generation_config.json +1 -1
  3. model.safetensors +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: apache-2.0
3
- base_model: t5-small
4
  tags:
5
  - generated_from_trainer
6
  metrics:
@@ -15,11 +15,11 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # translation_model
17
 
18
- This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 2.6186
21
- - Bleu: 0.3271
22
- - Gen Len: 18.2151
23
 
24
  ## Model description
25
 
@@ -44,20 +44,28 @@ The following hyperparameters were used during training:
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
- - num_epochs: 2
48
  - mixed_precision_training: Native AMP
49
 
50
  ### Training results
51
 
52
- | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
53
- |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|
54
- | 3.0275 | 1.0 | 4674 | 2.7024 | 0.2634 | 18.234 |
55
- | 2.9191 | 2.0 | 9348 | 2.6186 | 0.3271 | 18.2151 |
 
 
 
 
 
 
 
 
56
 
57
 
58
  ### Framework versions
59
 
60
- - Transformers 4.40.0
61
- - Pytorch 2.2.1+cu121
62
  - Datasets 2.19.0
63
  - Tokenizers 0.19.1
 
1
  ---
2
  license: apache-2.0
3
+ base_model: drewcurran/translation_model
4
  tags:
5
  - generated_from_trainer
6
  metrics:
 
15
 
16
  # translation_model
17
 
18
+ This model is a fine-tuned version of [drewcurran/translation_model](https://huggingface.co/drewcurran/translation_model) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 1.9529
21
+ - Bleu: 1.2244
22
+ - Gen Len: 18.0659
23
 
24
  ## Model description
25
 
 
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - num_epochs: 10
48
  - mixed_precision_training: Native AMP
49
 
50
  ### Training results
51
 
52
+ | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
53
+ |:-------------:|:-----:|:-----:|:---------------:|:------:|:-------:|
54
+ | 2.5538 | 1.0 | 4674 | 2.2471 | 0.7389 | 18.1428 |
55
+ | 2.4606 | 2.0 | 9348 | 2.1615 | 0.8787 | 18.1105 |
56
+ | 2.3877 | 3.0 | 14022 | 2.0981 | 0.963 | 18.0917 |
57
+ | 2.3433 | 4.0 | 18696 | 2.0538 | 1.0353 | 18.0855 |
58
+ | 2.2914 | 5.0 | 23370 | 2.0186 | 1.0902 | 18.0692 |
59
+ | 2.2565 | 6.0 | 28044 | 1.9939 | 1.1566 | 18.0789 |
60
+ | 2.2539 | 7.0 | 32718 | 1.9756 | 1.1935 | 18.0686 |
61
+ | 2.2262 | 8.0 | 37392 | 1.9628 | 1.1998 | 18.0759 |
62
+ | 2.2097 | 9.0 | 42066 | 1.9556 | 1.2209 | 18.0706 |
63
+ | 2.2021 | 10.0 | 46740 | 1.9529 | 1.2244 | 18.0659 |
64
 
65
 
66
  ### Framework versions
67
 
68
+ - Transformers 4.40.1
69
+ - Pytorch 2.3.0+cu121
70
  - Datasets 2.19.0
71
  - Tokenizers 0.19.1
generation_config.json CHANGED
@@ -2,5 +2,5 @@
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
5
- "transformers_version": "4.40.0"
6
  }
 
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
5
+ "transformers_version": "4.40.1"
6
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83f22924b83d3f9cef765cc1ab2838e103630017b703a44f94fcdcf3c1aef686
3
  size 242041896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9713fcd039199e4a5d23725c7d84daf19b237b7cd3fd25b183e0ba0c2b4c6bcf
3
  size 242041896