CoolFace
Modelpublic

distill-lab/train_dit02_initials_3cat_cls_3.5k_naflex

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

traindit02initials3catcls3.5knaflex

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

  • —Loss: 2.2541
  • —Accuracy: 0.6474
  • —Precision: 0.6451
  • —Recall: 0.6474
  • —F1: 0.6312
  • —Roc Auc: 0.7542

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: 5e-06
  • —trainbatchsize: 64
  • —evalbatchsize: 64
  • —seed: 1337
  • —optimizer: Use adamwtorch with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • —lrschedulertype: cosine
  • —num_epochs: 20.0
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossAccuracyPrecisionRecallF1Roc Auc
0.16062.08331000.29930.65300.65370.65300.63950.7772
0.05114.16672000.61500.61010.67700.61010.60380.7704
0.01046.253000.79660.65490.63600.65490.63940.7630
0.01148.33334000.99930.66040.65990.66040.64890.7559
0.001610.41675001.29270.65300.63500.65300.62670.7524
0.000212.56001.48760.62870.64000.62870.61620.7600
0.014.58337002.06750.64550.63360.64550.62710.7547
0.016.66678002.19740.64930.63900.64930.63050.7562
0.018.759002.25120.64930.64730.64930.63310.7541

Framework versions

  • —Transformers 4.56.2
  • —Pytorch 2.7.1+cu126
  • —Datasets 4.0.0
  • —Tokenizers 0.22.1