padam5699/pets-convnext-sagemaker

Pet breed classifier (37 cat and dog breeds)

Fine-tuned on the Oxford-IIIT Pet dataset (80/20 stratified split of all 7,349 images) with a SageMaker training job, for a Distributed Computing class demo.

base model	facebook/convnext-tiny-224
epochs	5
validation accuracy	94.1%
trained on	SageMaker training job pets-pytorch-2026-11-04-20-10-08-796 (ml.g6.xlarge)
billable seconds	202

Use it

from transformers import pipeline
classifier = pipeline("image-classification", model="padam5699/pets-convnext-sagemaker")
classifier("my_cat.jpg")

Breeds: Abyssinian, Bengal, Birman, Bombay, British Shorthair, Egyptian Mau, Maine Coon, Persian, Ragdoll, Russian Blue, Siamese, Sphynx, American Bulldog, American Pit Bull Terrier, Basset Hound, Beagle, Boxer, Chihuahua, English Cocker Spaniel, English Setter, German Shorthaired, Great Pyrenees, Havanese, Japanese Chin, Keeshond, Leonberger, Miniature Pinscher, Newfoundland, Pomeranian, Pug, Saint Bernard, Samoyed, Scottish Terrier, Shiba Inu, Staffordshire Bull Terrier, Wheaten Terrier, Yorkshire Terrier.

Dataset: Parkhi et al., "Cats and Dogs", CVPR 2012; images under CC BY-SA 4.0.