<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->
helsinki-opus-de-en-fine-tuned-wmt16-finetuned-src-to-trg
This model is a fine-tuned version of mariav/helsinki-opus-de-en-fine-tuned-wmt16 on the None dataset. It achieves the following results on the evaluation set:
- Loss: 0.8597
- Rouge1: 64.539
- Rouge2: 32.7634
- Rougel: 61.3523
- Rougelsum: 61.3758
- Gen Len: 23.9561
- Bleu-1: 64.1391
- Bleu-2: 45.1093
- Bleu-3: 32.4697
- Bleu-4: 24.2684
- Meteor: 0.5436
Model description
This model is a fine-tuned version of mariav/helsinki-opus-de-en-fine-tuned-wmt16 on Phoenix Weather dataset (PHOENIX-2014-T).
Intended uses & limitations
The purpose is Neural Machine Translation from German text into German Sign Glosses, which could be used for avatar generation within the Sign Language Production task.
Training and evaluation data
Phoenix Weather dataset (PHOENIX-2014-T)
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 5
- eval_batch_size: 5
- seed: 7575
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 5
Training results
Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len | Bleu-1 | Bleu-2 | Bleu-3 | Bleu-4 | Meteor |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
1.1513 | 1.0 | 1189 | 0.9604 | 61.8236 | 30.0156 | 58.9651 | 58.9484 | 22.8563 | 58.6480 | 40.5508 | 29.0090 | 21.2884 | 0.4961 |
0.9067 | 2.0 | 2378 | 0.8825 | 62.8824 | 30.8604 | 59.9543 | 59.9884 | 22.7564 | 60.5598 | 42.0443 | 29.9532 | 21.8711 | 0.5138 |
0.739 | 3.0 | 3567 | 0.8547 | 63.8251 | 31.6294 | 60.7141 | 60.7508 | 24.5219 | 62.6847 | 43.6395 | 31.1174 | 22.8704 | 0.5318 |
0.636 | 4.0 | 4756 | 0.8554 | 64.5308 | 32.6897 | 61.347 | 61.3929 | 22.7912 | 63.0309 | 44.4786 | 32.0956 | 23.8647 | 0.5369 |
0.5745 | 5.0 | 5945 | 0.8597 | 64.539 | 32.7634 | 61.3523 | 61.3758 | 23.9561 | 64.1391 | 45.1093 | 32.4697 | 24.2684 | 0.5436 |
Framework versions
- Transformers 4.30.1
- Pytorch 2.0.1+cu118
- Datasets 2.12.0
- Tokenizers 0.13.3