food-classifier-224 / README.md
Dricz's picture
Update README.md
03d4685 verified
---
license: apache-2.0
datasets:
- food101
language:
- en
metrics:
- accuracy
pipeline_tag: image-classification
---
# Food Classifier
This model is a fine-tuned version of google/vit-base-patch16-224-in21k on the food101 dataset. It achieves the following results on the evaluation set:
<!-- Provide a quick summary of what the model is/does. -->
- **Evaluation loss:** 0.7166455984115601
- **Accuracy:** 0.8753663366336634
## Model Details
A model that can detect 101 variety of food.
### Model Description
<!-- Provide a longer summary of what this model is. -->
- **Developed by:** Dricz
- **Model type:** Image classification
- **Language(s) (NLP):** English
- **Finetuned from model:** google/vit-base-patch16-224-in21k
## Training Details
### Training Data
- **training_loss: 1.7299224627936907**
- **train_runtime:** 3538
- **train_samples_per_second:** 21.409
- **train_steps_per_second:** 1.338
- **total_flos:** 5.8752267138432e+18
- **train_loss:** 1.7299224627936907
- **epoch:** 1.0
#### Training Hyperparameters
The following hyperparameters were used during training:
- **learning_rate:** 5e-05
- **train_batch_size**: 16
- **seed:** 42
- **optimizer:** Adam with betas=(0.9,0.999) and epsilon=1e-08
- **lr_scheduler_type:** linear
- **num_epochs:** 1
## Evaluation
<!-- This section describes the evaluation protocols and provides the results. -->
- **eval_loss:** 0.7166455984115601
- **eval_accuracy:** 0.8753663366336634
- **eval_runtime:** 446.9362
- **eval_steps_per_second:** 3.533
- **epoch:** 1.0