vit-ena24

This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 1.3273
  • Accuracy: 0.7539

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
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: linear
  • num_epochs: 2
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Accuracy
1.5119 0.1259 100 1.8353 0.5625
0.809 0.2519 200 1.4106 0.6396
0.6754 0.3778 300 1.5657 0.5771
0.5017 0.5038 400 1.3136 0.6865
0.2595 0.6297 500 1.2942 0.6865
0.243 0.7557 600 1.3563 0.6914
0.3432 0.8816 700 1.4268 0.6689
0.1115 1.0076 800 1.4286 0.6973
0.1615 1.1335 900 1.4697 0.6963
0.115 1.2594 1000 1.4701 0.7109
0.0656 1.3854 1100 1.4417 0.7217
0.1229 1.5113 1200 1.3150 0.7451
0.1064 1.6373 1300 1.3941 0.7432
0.0345 1.7632 1400 1.2879 0.7607
0.0587 1.8892 1500 1.3273 0.7539

Framework versions

  • Transformers 4.51.3
  • Pytorch 2.6.0+cu124
  • Datasets 3.6.0
  • Tokenizers 0.21.1
Downloads last month
3
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for mbiarreta/vit-ena24

Finetuned
(2476)
this model

Evaluation results