Image Classification

Classifies cat and dog images using a subset of the cats_vs_dogs 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

Note: during the training, I tried adjusting some of the above hyperparameters (like making the learning rate 0.1 as we have seen in class). But then the model could only classify cats and not dogs.