nlp-vietnamese / README.md
52100176-NguyenTrongDat's picture
Training complete
8420c66 verified
|
raw
history blame
1.73 kB
metadata
base_model: vinai/bartpho-syllable
tags:
  - vietnamese-model
  - generated_from_trainer
metrics:
  - sacrebleu
model-index:
  - name: nlp-vietnamese
    results: []

nlp-vietnamese

This model is a fine-tuned version of vinai/bartpho-syllable on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0525
  • Sacrebleu: 21.4023

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: 2e-05
  • train_batch_size: 12
  • eval_batch_size: 12
  • seed: 42
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 48
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 5
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Sacrebleu
No log 1.0 166 0.1972 13.9262
No log 2.0 332 0.0939 18.5618
No log 3.0 498 0.0612 20.4411
No log 4.0 664 0.0540 21.0027
No log 5.0 830 0.0525 21.4023

Framework versions

  • Transformers 4.40.2
  • Pytorch 2.2.1+cu121
  • Datasets 2.19.1
  • Tokenizers 0.19.1