food-classifier-224

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