This model is a fine-tuned version of microsoft/resnet-50 on the imagefolder dataset.
It achieves the following results on the evaluation set:
- Loss: 0.9690
- Accuracy: 0.8149
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.0001
- train_batch_size: 8
- eval_batch_size: 16
- seed: 42
- gradient_accumulation_steps: 2
- total_train_batch_size: 16
- 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: cosine
- num_epochs: 20
- label_smoothing_factor: 0.1
Training results
| Training Loss |
Epoch |
Step |
Validation Loss |
Accuracy |
| 2.2527 |
1.0 |
63 |
2.2759 |
0.3072 |
| 2.1819 |
2.0 |
126 |
2.1895 |
0.4458 |
| 1.9930 |
3.0 |
189 |
1.9899 |
0.5331 |
| 1.6148 |
4.0 |
252 |
1.6825 |
0.5693 |
| 1.3376 |
5.0 |
315 |
1.4247 |
0.6476 |
| 1.1109 |
6.0 |
378 |
1.2356 |
0.6928 |
| 0.9855 |
7.0 |
441 |
1.1114 |
0.7380 |
| 0.8771 |
8.0 |
504 |
1.0299 |
0.7861 |
| 0.7898 |
9.0 |
567 |
0.9490 |
0.8072 |
| 0.7633 |
10.0 |
630 |
0.9580 |
0.8373 |
| 0.6495 |
11.0 |
693 |
0.9483 |
0.8072 |
| 0.6381 |
12.0 |
756 |
0.9191 |
0.8253 |
| 0.6942 |
13.0 |
819 |
0.9159 |
0.8313 |
| 0.6283 |
14.0 |
882 |
0.8666 |
0.8464 |
| 0.6214 |
15.0 |
945 |
0.9109 |
0.8404 |
| 0.6299 |
16.0 |
1008 |
0.8847 |
0.8404 |
| 0.6216 |
17.0 |
1071 |
0.8869 |
0.8343 |
| 0.5789 |
18.0 |
1134 |
0.8708 |
0.8614 |
| 0.6125 |
19.0 |
1197 |
0.9240 |
0.8343 |
| 0.6005 |
20.0 |
1260 |
0.9022 |
0.8464 |
Framework versions
- Transformers 5.16.1
- Pytorch 2.11.0+cu128
- Datasets 4.8.5
- Tokenizers 0.23.1