CoolFace
Modelpublic

crangana/trained-gender

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
2likes158downloads
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. -->

trained-gender

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

  • Loss: 0.2437
  • Accuracy: 0.8986

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.0002
  • trainbatchsize: 16
  • evalbatchsize: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • num_epochs: 4

Training results

Training LossEpochStepValidation LossAccuracy
0.42770.1810000.40540.8089
0.3150.3720000.34870.8318
0.30820.5530000.30520.8633
0.32350.7440000.28990.8684
0.25050.9250000.26930.8785
0.24841.1160000.25470.8889
0.19331.2970000.25210.8901
0.14971.4880000.24430.8929
0.3261.6690000.24060.8958
0.2151.84100000.23810.9007
0.20352.03110000.24370.8986

Framework versions

  • Transformers 4.34.0
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.5
  • Tokenizers 0.14.0