Downloads · 30 days
3
1% of all-time downloads
my_awesome_food_model
ashleyradford/my_awesome_food_model
my_awesome_food_model is a image classification model from ashleyradford. Use it when you need a label for an image. It is set up for transformers.
Classifies food images using a subset of the food101 dataset.<br Uses PyTorch for the preprocessing, training, and inference.
All-time downloads
475
Public
Repo size
1.4 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin344 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- Model type
- vit
- Access
- Public
- Created
- May 18, 2023
- Updated
- May 18, 2023
- SHA
- d7a384ee
Datasets
Model card
From the Hugging Face model README
Image Classification
Classifies food images using a subset of the food101 dataset.<br> Uses PyTorch for the preprocessing, training, and inference.
output_dir="cats_vs_dogs_model"
remove_unused_columns=False
evaluation_strategy="epoch"
save_strategy="epoch"
learning_rate=5e-5
per_device_train_batch_size=16
gradient_accumulation_steps=4
per_device_eval_batch_size=16
num_train_epochs=3
warmup_ratio=0.1
logging_steps=10
load_best_model_at_end=True
metric_for_best_model="accuracy"
push_to_hub=True
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- Languages
- en
- Created
- May 18, 2023
- Updated
- May 18, 2023