CoolFace
Modelpublic

HorcruxNo13/swiftformer-xs

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes21downloads
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. -->

swiftformer-xs

This model is a fine-tuned version of MBZUAI/swiftformer-xs on the imagefolder dataset. It achieves the following results on the evaluation set:

  • Loss: 0.6833
  • Accuracy: 0.57
  • Precision: 0.5995
  • Recall: 0.57
  • F1 Score: 0.5828

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-05
  • trainbatchsize: 64
  • evalbatchsize: 64
  • seed: 42
  • gradientaccumulationsteps: 4
  • totaltrainbatch_size: 256
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • lrschedulerwarmup_ratio: 0.1
  • num_epochs: 30

Training results

Training LossEpochStepValidation LossAccuracyPrecisionRecallF1 Score
No log1.040.67130.62920.64540.62920.6365
No log2.080.71420.4750.61550.4750.5020
No log3.0120.72980.4250.60260.4250.4435
No log4.0160.73890.47920.64080.47920.5023
No log5.0200.74270.47920.64080.47920.5023
No log6.0240.72350.50830.64240.50830.5348
No log7.0280.68930.58750.66870.58750.6107
0.69818.0320.68160.60420.68470.60420.6264
0.69819.0360.68660.60420.68880.60420.6266
0.698110.0400.70050.5750.67510.5750.5996
0.698111.0440.71270.5250.65540.5250.5510
0.698112.0480.70980.53330.65950.53330.5593
0.698113.0520.71260.52080.65790.52080.5463
0.698114.0560.71140.52920.65750.52920.5551
0.665615.0600.69080.56670.67120.56670.5917
0.665616.0640.68040.58330.67490.58330.6073
0.665617.0680.68060.59580.68080.59580.6188
0.665618.0720.68840.55830.66290.55830.5838
0.665619.0760.68210.57080.66470.57080.5955
0.665620.0800.66630.60420.68060.60420.6261
0.665621.0840.67170.60.67870.60.6223
0.665622.0880.66820.60830.68260.60830.6299
0.644323.0920.66830.61670.69460.61670.6381
0.644324.0960.67330.60.69110.60.6230
0.644325.01000.66470.60830.68660.60830.6302
0.644326.01040.67290.60830.69070.60830.6305
0.644327.01080.67400.60420.69300.60420.6268
0.644328.01120.68090.59170.69160.59170.6153
0.644329.01160.67780.60420.70170.60420.6270
0.631330.01200.67940.59580.69350.59580.6192

Framework versions

  • Transformers 4.33.3
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.5
  • Tokenizers 0.13.3