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