|
--- |
|
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 |
|
|