CoolFace
Modelpublic

Foxasdf/EfficientNetV2_Small_v1

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

EfficientNetV2Smallv1

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

  • Loss: 0.0340
  • Accuracy: 0.9935
  • Precision: 0.9981
  • Recall: 0.9878
  • F1: 0.9929
  • Tp: 1618
  • Tn: 1907
  • Fp: 3
  • Fn: 20

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: 64
  • evalbatchsize: 64
  • 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: 442
  • num_epochs: 20
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossAccuracyPrecisionRecallF1TpTnFpFn
0.19951.02220.13490.96280.95750.96210.9598157618407062
0.14422.04440.09400.97890.99560.95850.976715701903768
0.16253.06660.08270.98370.99250.97190.9821159218981246
0.15924.08880.09260.97520.97080.97560.9732159818624840
0.11005.011100.05440.98760.99500.97800.986516021902836
0.14976.013320.06350.98680.98770.98350.9856161118902027
0.11257.015540.04850.98960.99570.98170.988616081903730
0.12028.017760.07740.97940.97400.98170.9778160818674330
0.10319.019980.05070.98930.99380.98290.9883161019001028
0.121110.022200.04340.99150.99750.98410.990816121906426
0.123911.024420.04000.99180.99750.98470.991116131906425
0.106612.026640.04030.99270.99880.98530.992016141908224
0.106513.028860.03630.99270.99940.98470.992016131909125
0.107414.031080.03780.99300.99880.98600.992316151908223
0.112815.033300.03270.99240.99810.98530.991716141907324
0.096316.035520.03090.99300.99880.98600.992316151908223
0.137917.037740.03660.99270.99690.98720.992016171905521
0.107018.039960.03310.99300.99810.98660.992316161907322
0.133219.042180.03430.99300.99810.98660.992316161907322
0.129420.044400.03400.99350.99810.98780.992916181907320

Framework versions

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