CIFAR-10 prediction with PyTorch

Let's explore the CIFAR-10 dataset, but with PyTorch. We are going to use a program that is based on the official PyTorch documentation for CPUs. I've extensively modified this program for GPUs for both training and prediction.

But, first, we must include a copy of the PyTorch license from GitHub. Through a series code blocks, we can now take a look at the GitHub license:

From PyTorch: Copyright (c) 2016- Facebook, Inc (Adam Paszke) Copyright (c) 2014- Facebook, Inc (Soumith Chintala) Copyright (c) 2011-2014 Idiap Research Institute (Ronan Collobert) Copyright (c) 2012-2014 Deepmind Technologies (Koray Kavukcuoglu) Copyright (c) 2011-2012 NEC Laboratories America (Koray Kavukcuoglu) Copyright (c) 2011-2013 ...

Get Hands-On GPU Computing with Python now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.