CoolFace
Modelpublic

priyankloco/resnet-152-finetuned_resnet152-adam-optimizer5e-4-autotags

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

resnet-152-finetuned_resnet152-adam-optimizer5e-4-autotags

This model is a fine-tuned version of microsoft/resnet-152 on the imagefolder dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.2399
  • —Accuracy: 0.9305

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

Training results

Training LossEpochStepValidation LossAccuracy
2.40090.99652.14140.3971
0.92011.991300.81230.7210
0.75752.991950.57300.8124
0.47923.992600.41660.8648
0.42534.993250.38110.8810
0.33315.993900.42900.8705
0.23476.994550.46000.8952
0.17327.995200.30180.8924
0.17778.995850.48510.8914
0.12989.996500.29410.92
0.116410.997150.39150.9095
0.128411.997800.37010.9152
0.098612.998450.34160.9171
0.094413.999100.31450.9210
0.092914.999750.26770.9229
0.101415.9910400.27450.9295
0.097116.9911050.29320.9267
0.069117.9911700.21740.9333
0.055718.9912350.22330.9324
0.0619.9913000.23990.9305

Framework versions

  • —Transformers 4.25.1
  • —Pytorch 1.13.1+cu117
  • —Datasets 2.11.0
  • —Tokenizers 0.13.2