Team Ai
Modelpublic

davelotito/donut_experiment_bayesian_trial_6

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes9downloads
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. -->

donutexperimentbayesiantrial6

This model is a fine-tuned version of naver-clova-ix/donut-base on the None dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.5515
  • —Bleu: 0.0683
  • —Precisions: [0.8127572016460906, 0.7412587412587412, 0.6854838709677419, 0.638095238095238]
  • —Brevity Penalty: 0.0954
  • —Length Ratio: 0.2985
  • —Translation Length: 486
  • —Reference Length: 1628
  • —Cer: 0.7532
  • —Wer: 0.8274

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.00016063260663724173
  • —trainbatchsize: 1
  • —evalbatchsize: 1
  • —seed: 42
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 2
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 4
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossBleuPrecisionsBrevity PenaltyLength RatioTranslation LengthReference LengthCerWer
0.32761.02530.66720.0589[0.76875, 0.6737588652482269, 0.6092896174863388, 0.5436893203883495]0.09150.294848016280.75860.8473
0.20082.05060.57800.0662[0.7905544147843943, 0.7069767441860465, 0.6595174262734584, 0.6107594936708861]0.09600.299148716280.75590.8374
0.13563.07590.53550.0651[0.8238993710691824, 0.7452380952380953, 0.6942148760330579, 0.6535947712418301]0.08950.293047716280.75800.8299
0.03944.010120.55150.0683[0.8127572016460906, 0.7412587412587412, 0.6854838709677419, 0.638095238095238]0.09540.298548616280.75320.8274

Framework versions

  • —Transformers 4.40.0
  • —Pytorch 2.1.0
  • —Datasets 2.18.0
  • —Tokenizers 0.19.1