CoolFace
Modelpublic

maud-dr/model_2_stage2-seed_42

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes11downloads
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. -->

model2stage2-seed_42

This model is a fine-tuned version of maud-dr/model_2_stage1 on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 2.9477
  • Precision: 0.6407
  • Recall: 0.6848
  • F1: 0.6620

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.0003
  • trainbatchsize: 8
  • evalbatchsize: 8
  • seed: 42
  • optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • lrschedulertype: linear
  • num_epochs: 15

Training results

Training LossEpochStepValidation LossPrecisionRecallF1
0.64941.04470.77330.56730.77900.6565
0.57912.08940.63120.60760.80800.6936
0.50433.013410.70730.61710.70650.6588
0.41174.017880.79080.61980.750.6787
0.35715.022351.04480.60170.750.6677
0.34216.026821.13550.61760.71380.6622
0.27657.031291.24170.65120.66300.6571
0.23048.035761.56470.63290.72460.6757
0.14359.040232.07340.63780.72100.6769
0.127510.044702.04320.64030.70290.6701
0.136111.049172.24700.64030.70290.6701
0.065812.053642.65460.63140.71380.6701
0.062613.058112.70800.64310.69200.6667
0.039614.062582.94360.62010.69200.6541
0.047515.067052.94770.64070.68480.6620

Framework versions

  • Transformers 4.52.0.dev0
  • Pytorch 2.7.0+cu126
  • Datasets 3.6.0
  • Tokenizers 0.21.1