Create and train a PyTorch model for digit classification

About this Learning Path

Who is this for?

This is an introductory topic for software developers interested in learning how to use PyTorch to create and train a feedforward neural network for digit classification.

What will you learn?

Upon completion of this learning path, you will be able to:

• Prepare a PyTorch development environment.
• Download and prepare the MNIST dataset.
• Create a neural network architecture using PyTorch.
• Train a neural network using PyTorch.

Prerequisites

Before starting, you will need the following:

• A computer that can run Python3 and Visual Studio Code. The OS can be Windows, Linux, or macOS.