Exciting news! Gradient has launched a FREE GPU plan. Read More
Project Details

PyTorch Tutorial: Training a Classifier

Learn how to train an image classifier using the torchvision package in PyTorch

The Gradient Team


PyTorch has quickly become today's most popular deep learning platform. In this tutorial by Soumith Chintala, one of the creators of PyTorch, you'll learn how to train an image classifier on the CIFAR10 dataset using the torchvision package. Specifically, we'll see how to:

  • Load and normalize the CIFAR10 training and test datasets using torchvision
  • Define a Convolutional Neural Network
  • Define a loss function
  • Train the network on the training data
  • Test the network on the test data