CoolFace
Modelpublic

aynumosir/mt5-base-ainu

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes364downloads
Model Card

<!-- 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. -->

mt5-base-ainu

This model is a fine-tuned version of google/mt5-base on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 2.1105
  • Bleu: 37.4939

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: 0.0005
  • trainbatchsize: 16
  • evalbatchsize: 16
  • seed: 42
  • gradientaccumulationsteps: 2
  • totaltrainbatch_size: 32
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • lrschedulerwarmup_ratio: 0.06
  • num_epochs: 20

Training results

Training LossEpochStepValidation LossBleu
2.12671.093411.802620.8450
1.64082.0186821.470626.7109
1.40983.0280231.349430.7048
1.25464.0373641.291032.5056
1.1245.0467051.261733.7060
1.00486.0560461.257834.5824
0.88727.0653871.263935.1029
0.81038.0747281.295535.7998
0.72989.0840691.328435.8310
0.649410.0934101.378036.3268
0.569611.01027511.434336.2494
0.514812.01120921.495736.8702
0.448713.01214331.551136.8981
0.394114.01307741.623536.8809
0.343215.01401151.695737.0269
0.302316.01494561.793537.1839
0.261417.01587971.861937.1935
0.226718.01681381.948537.4170
0.199619.01774792.034837.3585
0.174620.01868202.110537.4939

Framework versions

  • Transformers 4.40.1
  • Pytorch 2.1.2
  • Datasets 2.19.0
  • Tokenizers 0.19.1