Skip to content

heatherbshapiro/pycon-canada

Repository files navigation

Welcome to Pycon Canada!

In this tutorial, you will learn how to train your own image classification model using transfer learning. The Azure Machine Learning python SDK's PyTorch estimator enables you to easily submit PyTorch training jobs for both single-node and distributed runs on Azure compute. The model is trained to classify dog breeds using a pretrained ResNet18 model that has been trained on the Stanford Dog dataset. This dataset has been built using images and annotation from ImageNet for the task of fine-grained image categorization. For time, we will use a subset of this dataset which includes 10 dog breeds.

You can view the subset of the data used here.

Please refer to the dog-breed-classifier.ipynb notebook for instructions.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published