CoolFace
Modelpublic

carlosleao/FER-Facial-Expression-Recognition

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes39downloads
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. -->

FER-Facial-Expression-Recognition

This model is a fine-tuned version of motheecreator/vit-Facial-Expression-Recognition on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4710
  • Accuracy: 0.8474

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: 3e-05
  • trainbatchsize: 32
  • evalbatchsize: 32
  • seed: 42
  • gradientaccumulationsteps: 8
  • totaltrainbatch_size: 256
  • optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • lrschedulertype: cosine
  • lrschedulerwarmup_steps: 1000
  • num_epochs: 10

Training results

Training LossEpochStepValidation LossAccuracy
1.88680.89591001.76380.5923
1.22771.79622001.10920.7253
0.84142.69653000.81050.8041
0.70763.59694000.67460.8256
0.60794.49725000.61110.8287
0.56245.39756000.55290.8379
0.52546.29797000.52660.8399
0.47847.19828000.49780.8433
0.46348.09859000.48440.8458
0.43058.994410000.47100.8474
0.39959.894711000.43810.8564

Framework versions

  • Transformers 4.46.2
  • Pytorch 2.5.1+cu124
  • Datasets 3.1.0
  • Tokenizers 0.20.3