
# 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
```
apple_pie baby_back_ribs bruschetta waffles caesar_salad cannoli caprese_salad carrot_cake ceviche cheesecake cheese_plate chicken_curry chicken_quesadilla baklava chicken_wings chocolate_cake chocolate_mousse churros clam_chowder club_sandwich crab_cakes creme_brulee croque_madame cup_cakes beef_carpaccio deviled_eggs donuts dumplings edamame eggs_benedict escargots falafel filet_mignon fish_and_chips foie_gras beef_tartare french_fries french_onion_soup french_toast fried_calamari fried_rice frozen_yogurt garlic_bread gnocchi greek_salad grilled_cheese_sandwich beet_salad grilled_salmon guacamole gyoza hamburger hot_and_sour_soup hot_dog huevos_rancheros hummus ice_cream lasagna beignets lobster_bisque lobster_roll_sandwich macaroni_and_cheese macarons miso_soup mussels nachos omelette onion_rings oysters bibimbap pad_thai paella pancakes panna_cotta peking_duck pho pizza pork_chop poutine prime_rib bread_pudding pulled_pork_sandwich ramen ravioli red_velvet_cake risotto samosa sashimi scallops seaweed_salad shrimp_and_grits breakfast_burrito spaghetti_bolognese spaghetti_carbonara spring_rolls steak strawberry_shortcake sushi tacos takoyaki tiramisu tuna_tartare