CoolFace
Modelpublic

Foxasdf/mobilenet_v4_small

sourceHugging Faceapache-2.0updated 7mo 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. -->

mobilenetv4small

This model is a fine-tuned version of timm/mobilenetv4_conv_small.e2400_r224_in1k on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0668
  • Accuracy: 0.9817
  • Precision: 0.9870
  • Recall: 0.9731
  • F1: 0.9800
  • Tp: 1594
  • Tn: 1889
  • Fp: 21
  • Fn: 44

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.0001
  • trainbatchsize: 128
  • evalbatchsize: 128
  • seed: 42
  • optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lrschedulertype: linear
  • lrschedulerwarmup_steps: 220
  • num_epochs: 20
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossAccuracyPrecisionRecallF1TpTnFpFn
0.24581.01110.36150.83930.76510.94080.84391541143747397
0.22612.02220.28360.87150.80500.95240.87251560153237878
0.22613.03330.36700.80890.71640.97010.82421589128162949
0.21504.04440.15810.95320.94500.95420.9496156318199175
0.18535.05550.14920.95550.94470.95970.9522157218189266
0.15096.06660.12600.96450.96210.96090.9615157418486264
0.15687.07770.09520.97770.98210.96950.9757158818812950
0.12868.08880.08780.97550.98440.96210.9731157618852562
0.17429.09990.09500.97440.97370.97070.9722159018674348
0.162310.011100.09440.97210.97180.96760.9697158518644653
0.155011.012210.08350.98080.99310.96520.9789158118991157
0.152612.013320.17510.94590.91410.97440.94331596176015042
0.117513.014430.06380.98340.99250.97130.9818159118981247
0.134914.015540.07540.97940.98390.97130.9776159118842647
0.112115.016650.08220.97940.98510.97010.9775158918862449
0.131916.017760.07470.98080.98340.97500.9792159718832741
0.136717.018870.06460.98280.99070.97190.9812159218951546
0.129718.019980.06900.98200.98580.97500.9804159718872341
0.121819.021090.06920.98110.98640.97250.9794159318882245
0.138020.022200.06680.98170.98700.97310.9800159418892144

Framework versions

  • Transformers 5.2.0
  • Pytorch 2.9.0+cu126
  • Datasets 4.0.0
  • Tokenizers 0.22.2