|
--- |
|
datasets: |
|
- food101 |
|
language: |
|
- en |
|
metrics: |
|
- accuracy |
|
library_name: transformers |
|
--- |
|
|
|
# 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 |
|
``` |
