Skip to main content

Sloth or Pastry? Using PyTorch and Deep Learning for Image Classification


In the age of deep learning, data scientists and machine learning engineers seldom create and train neural networks from scratch. A big chunk of what goes into performing a machine learning task, however, is collecting, preparing, and loading data to feed into a model. 

In this session, using DataCamp Workspace we'll learn all about loading custom datasets into PyTorch and using transfer learning to perform an image processing task using a mostly-pretrained model, which we'll fine-tune.

We'll be using computer vision to answer the internet-popular question: is it a sloth or a pain au chocolat? This is a binary image classification task. 

Key takeaways

  • Machine learning engineers seldom train models from scratch
  • Learning to load data is an important building block for deep learning in PyTorch
  • Models trained on billions of images are available at our disposal for transfer learning

Solution code in DataCamp Workspace

Start Code Along
Maham Khan Headshot
Maham Khan

Senior Data Science Content Developer

View More Webinars

Hands-on learning experience

Companies using DataCamp achieve course completion rates 6X higher than traditional online course providers

Learn More

Upskill your teams in data science and analytics

Learn More

Join 2,500+ companies and 80% of the Fortune 1000 who use DataCamp to upskill their teams.

Don’t just take our word for it.