hungngo04's picture
update model card README.md
8179c98
metadata
license: apache-2.0
tags:
  - generated_from_trainer
metrics:
  - bleu
model-index:
  - name: cluster_to_text_t5_base_test
    results: []

cluster_to_text_t5_base_test

This model is a fine-tuned version of t5-small on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 1.7184
  • Bleu: 12.1051
  • Gen Len: 10.6659

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: 16
  • eval_batch_size: 16
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 3

Training results

Training Loss Epoch Step Validation Loss Bleu Gen Len
2.4089 1.0 4678 2.1095 3.8133 11.5338
2.1256 2.0 9356 1.8032 9.8148 10.7901
2.0173 3.0 14034 1.7184 12.1051 10.6659

Framework versions

  • Transformers 4.30.2
  • Pytorch 2.0.1+cu118
  • Datasets 2.13.1
  • Tokenizers 0.13.3