YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
UpsideDownClassifier
This classifier was trained using the auto-cats-and-dogs dataset. It was trained over 5 epochs using a pretrained resent18.
The configuration for the model was
config = {
"batch_size": 64,
"num_epochs": 5,
"lr": 0.005,
"betas": (0.9, 0.999),
"eps": 1e-6,
"lr": 8e-3,
"do_eval": True
}
Traning Plots
We can see in the figures below the training plots for accuracy and the loss in both, training and validation sets.
Accuracy Plot
Loss Plot
Some Results
Evaluating on the Test Set, we obtain:
- Accuracy = 0.9696
A batch with some missclassifications can be seen in the picture below.
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
HF Inference API was unable to determine this model's library.