CoolFace
Modelpublic

robertsw/aesthetics_v2

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

aesthetics_v2

This model is a fine-tuned version of facebook/dinov2-large on the imagefolder dataset. It achieves the following results on the evaluation set:

  • Loss: 1.6501
  • Accuracy: 0.5581

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: 3

Training results

Training LossEpochStepValidation LossAccuracy
1.14650.17201.68600.5313
1.27030.34401.84120.5014
1.31520.52601.82000.5042
1.23130.69801.79710.5112
1.34760.861001.76490.5100
1.25971.031201.74540.5175
1.00941.21401.73560.5257
0.97431.371601.70740.5352
1.02091.551801.73310.5322
1.06921.722001.73700.5331
1.05561.892201.67880.5487
0.86342.062401.66440.5536
0.792.232601.68480.5531
0.79162.42801.67610.5528
0.74542.583001.65200.5534
0.74972.753201.63370.5554
0.75372.923401.65010.5581

Framework versions

  • Transformers 4.38.2
  • Pytorch 2.2.0
  • Datasets 2.17.1
  • Tokenizers 0.15.2