kinleyrabgay's picture
Update README.md
5628cd9 verified
metadata
library_name: transformers
license: cc-by-nc-4.0
base_model: kinleyrabgay/nllb-200-600M-dzo-eng
tags:
  - generated_from_trainer
metrics:
  - bleu
model-index:
  - name: nllb-200-600M-dzo-eng-30k
    results: []

nllb-200-600M-dzo-eng-30k

This model is a fine-tuned version of kinleyrabgay/nllb-200-600M-dzo-eng on kinleyrabgay/dz_to_en dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0678
  • Bleu: 63.0623

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 3e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • gradient_accumulation_steps: 2
  • total_train_batch_size: 8
  • optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: linear
  • num_epochs: 5
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Bleu
0.0688 1.0 2500 0.0644 62.0343
0.0543 2.0 5000 0.0634 62.8454
0.0444 3.0 7500 0.0643 63.3940
0.0377 4.0 10000 0.0658 63.4416
0.0315 5.0 12500 0.0678 63.0623

Framework versions

  • Transformers 4.52.4
  • Pytorch 2.6.0+cu124
  • Datasets 3.6.0
  • Tokenizers 0.21.1