Home » Tutorials » PyTorch

Training a Classifier

An adaptation of training a classifier tutorial using Habana Gaudi AI processors. In this tutorial, we will train a Convolutional Neural Network (CNN) to classify images. We will start with CPU training, and then continue to make the needed changes for training the model on Gaudi HPU. Handling data Generally, when you have to deal with … Read more

Quick Start

An adaptation of PyTorch Quickstart tutorial using Habana Gaudi AI processors. This tutorial, demonstrates how to migrate an existing PyTorch workload to Gaudi. The migration requires only loading the Habana PyTorch plugin library. This section runs through the API for common tasks in machine learning. Working with data PyTorch has two primitives to work with data: torch.utils.data.DataLoader and torch.utils.data.Dataset. Dataset stores … Read more

Sign up for the latest Habana developer news, events, training, and updates.