--- 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: - **Evaluation loss:** 0.7166455984115601 - **Accuracy:** 0.8753663366336634 ## Model Details A model that can detect 101 variety of food. ### Model Description - **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 - **eval_loss:** 0.7166455984115601 - **eval_accuracy:** 0.8753663366336634 - **eval_runtime:** 446.9362 - **eval_steps_per_second:** 3.533 - **epoch:** 1.0