Image Classification
Classifies food images using a subset of the food101 dataset.
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
- Downloads last month
- 2
This model does not have enough activity to be deployed to Inference API (serverless) yet.
Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.