Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

YouTube

Implementing and Training a Neural Network with PyTorch

Valerio Velardo - The Sound of AI via YouTube

Overview

Build and train a feed-forward neural network using PyTorch in this 37-minute tutorial. Develop a classifier to identify digits in the MNIST dataset while learning data management techniques with PyTorch datasets and data loaders. Follow a step-by-step implementation process, starting with PyTorch installation and dataset download. Progress through implementing a data loader, designing a feed-forward network, and creating a training loop. Conclude by training and storing the model, gaining practical experience in neural network development with PyTorch.

Syllabus

Intro
Installing PyTorch with pip
Step-by-step implementation overview
Download datasets
Implementing a data loader
Implementing a feed forward network
Implementing the training loop
Training and storing our model
Coming up next + outro

Taught by

Valerio Velardo - The Sound of AI

Reviews

Start your review of Implementing and Training a Neural Network with PyTorch

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Someone learning on their laptop while sitting on the floor.