You can deploy this model and start earning money today!

mluger/
LinearHead

transformers

safetensors

generated_from_trainer

imagefolder

google/vit-base-patch16-224-in21k

finetune

apache-2.0

endpoints_compatible

Views

No views yet

LinearHead

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

Loss: 0.8704
Accuracy: 0.6835

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
train_batch_size: 32
eval_batch_size: 32
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: cosine
lr_scheduler_warmup_steps: 200
num_epochs: 8

Training results

Training Loss	Epoch	Step	Validation Loss	Accuracy
1.5773	0.2972	200	1.6732	0.4634
1.2645	0.5944	400	1.2946	0.5648
1.1417	0.8915	600	1.1658	0.5922
1.1329	1.1887	800	1.1074	0.6042
1.0759	1.4859	1000	1.0992	0.6027
1.0251	1.7831	1200	1.0243	0.6339
0.9703	2.0802	1400	1.0028	0.6381
0.9723	2.3774	1600	0.9844	0.6389
0.9315	2.6746	1800	0.9719	0.6474
0.9436	2.9718	2000	0.9533	0.6499
0.8974	3.2689	2200	0.9357	0.6637
0.8836	3.5661	2400	0.9292	0.6590
0.8598	3.8633	2600	0.9211	0.6658
0.8131	4.1605	2800	0.9068	0.6722
0.7545	4.4577	3000	0.8970	0.6716
0.8033	4.7548	3200	0.8866	0.6739
0.7536	5.0520	3400	0.8895	0.6751
0.7624	5.3492	3600	0.8875	0.6746
0.7371	5.6464	3800	0.8793	0.6799
0.7554	5.9435	4000	0.8760	0.6843
0.6829	6.2407	4200	0.8749	0.6824
0.745	6.5379	4400	0.8717	0.6835
0.7033	6.8351	4600	0.8735	0.6842
0.6657	7.1322	4800	0.8717	0.6854
0.6527	7.4294	5000	0.8711	0.6849
0.6186	7.7266	5200	0.8704	0.6835

Framework versions

Transformers 4.51.1
Pytorch 2.6.0+cu124
Datasets 3.5.0
Tokenizers 0.21.1

mluger/LinearHead

LinearHead

Model description

Intended uses & limitations

Training and evaluation data

Training procedure

Training hyperparameters

Training results

Framework versions

mluger/
LinearHead