vit-base-patch16-384-finetuned-humid-classes-29

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

• Loss: 0.1751
• Accuracy: 0.9706
• F1 Macro: 0.9481
• Precision Macro: 0.9667
• Recall Macro: 0.9444
• Precision Dry: 1.0
• Recall Dry: 1.0
• F1 Dry: 1.0
• Precision Firm: 1.0
• Recall Firm: 1.0
• F1 Firm: 1.0
• Precision Humid: 1.0
• Recall Humid: 1.0
• F1 Humid: 1.0
• Precision Lump: 1.0
• Recall Lump: 1.0
• F1 Lump: 1.0
• Precision Moist: 0.8
• Recall Moist: 1.0
• F1 Moist: 0.8889
• Precision Rockies: 1.0
• Recall Rockies: 0.6667
• F1 Rockies: 0.8

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
• train_batch_size: 16
• eval_batch_size: 16
• seed: 42
• gradient_accumulation_steps: 4
• total_train_batch_size: 64
• optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
• lr_scheduler_type: linear
• lr_scheduler_warmup_ratio: 0.1
• num_epochs: 50

Training results

Training Loss	Epoch	Step	Validation Loss	Accuracy	F1 Macro	Precision Macro	Recall Macro	Precision Dry	Recall Dry	F1 Dry	Precision Firm	Recall Firm	F1 Firm	Precision Humid	Recall Humid	F1 Humid	Precision Lump	Recall Lump	F1 Lump	Precision Moist	Recall Moist	F1 Moist	Precision Rockies	Recall Rockies	F1 Rockies
No log	1.0	3	1.6572	0.2647	0.1201	0.1012	0.1710	0.0	0.0	0.0	0.4167	0.4545	0.4348	0.1905	0.5714	0.2857	0.0	0.0	0.0	0.0	0.0	0.0	0.0	0.0	0.0
No log	2.0	6	1.3471	0.5588	0.2952	0.3643	0.3611	0.0	0.0	0.0	0.6471	1.0	0.7857	0.5385	1.0	0.7	1.0	0.1667	0.2857	0.0	0.0	0.0	0.0	0.0	0.0
No log	3.0	9	0.9992	0.8529	0.8164	0.8824	0.7917	1.0	0.6667	0.8	0.9167	1.0	0.9565	0.7778	1.0	0.875	1.0	0.6667	0.8	0.6	0.75	0.6667	1.0	0.6667	0.8
1.4764	4.0	12	0.6843	0.8824	0.8418	0.8963	0.8194	1.0	0.6667	0.8	1.0	1.0	1.0	0.7778	1.0	0.875	1.0	0.8333	0.9091	0.6	0.75	0.6667	1.0	0.6667	0.8
1.4764	5.0	15	0.3771	0.9412	0.9205	0.9444	0.9167	0.75	1.0	0.8571	0.9167	1.0	0.9565	1.0	1.0	1.0	1.0	0.8333	0.9091	1.0	1.0	1.0	1.0	0.6667	0.8
1.4764	6.0	18	0.2362	0.9412	0.9205	0.9444	0.9167	0.75	1.0	0.8571	0.9167	1.0	0.9565	1.0	1.0	1.0	1.0	0.8333	0.9091	1.0	1.0	1.0	1.0	0.6667	0.8
0.4965	7.0	21	0.1751	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.4965	8.0	24	0.1442	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.4965	9.0	27	0.1957	0.9412	0.9139	0.9375	0.9028	1.0	1.0	1.0	1.0	1.0	1.0	0.875	1.0	0.9333	1.0	1.0	1.0	0.75	0.75	0.75	1.0	0.6667	0.8
0.0671	10.0	30	0.1178	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0671	11.0	33	0.1203	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0671	12.0	36	0.1498	0.9412	0.9139	0.9375	0.9028	1.0	1.0	1.0	1.0	1.0	1.0	0.875	1.0	0.9333	1.0	1.0	1.0	0.75	0.75	0.75	1.0	0.6667	0.8
0.0671	13.0	39	0.0931	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0138	14.0	42	0.0909	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0138	15.0	45	0.1133	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0138	16.0	48	0.1204	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0032	17.0	51	0.1193	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0032	18.0	54	0.1181	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0032	19.0	57	0.1174	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0018	20.0	60	0.1129	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0018	21.0	63	0.1099	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0018	22.0	66	0.1072	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0018	23.0	69	0.1028	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0013	24.0	72	0.0990	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0013	25.0	75	0.0963	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0013	26.0	78	0.0952	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0011	27.0	81	0.0959	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0011	28.0	84	0.0974	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0011	29.0	87	0.0999	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0009	30.0	90	0.1026	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0009	31.0	93	0.1046	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0009	32.0	96	0.1065	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0009	33.0	99	0.1074	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0008	34.0	102	0.1080	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0008	35.0	105	0.1077	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0008	36.0	108	0.1074	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0007	37.0	111	0.1073	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0007	38.0	114	0.1068	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0007	39.0	117	0.1065	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	40.0	120	0.1064	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	41.0	123	0.1065	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	42.0	126	0.1066	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	43.0	129	0.1067	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	44.0	132	0.1067	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	45.0	135	0.1068	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0007	46.0	138	0.1068	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0006	47.0	141	0.1068	0.9706	0.9429	0.9583	0.9444	0.75	1.0	0.8571	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.6667	0.8
0.0006	48.0	144	0.1069	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0006	49.0	147	0.1069	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8
0.0006	50.0	150	0.1069	0.9706	0.9481	0.9667	0.9444	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	1.0	0.8	1.0	0.8889	1.0	0.6667	0.8

Framework versions

• Transformers 4.57.1
• Pytorch 2.9.0+cu126
• Datasets 4.0.0
• Tokenizers 0.22.0