Create and train a PyTorch model for digit classification: Next Steps

Developer Hub

Learning Paths

Learning-Paths

Servers and Cloud Computing

Create and train a PyTorch model for digit classification

Next Steps

Create and train a PyTorch model for digit classification

Introduction

Prepare a PyTorch development environment

Create a PyTorch model for MNIST

About PyTorch model training

Perform training and save the model

Use the model for inference

Understand inference on Android

Create an Android application

Prepare Test Data

Run the Application

Optimizing neural network models in PyTorch

Create an optimized PyTorch model for MNIST

Run optimization

Update the Android application

Review

Next Steps

Log an issue

Fork and edit

Discuss on Discord

Create and train a PyTorch model for digit classification

Introduction

Prepare a PyTorch development environment

Create a PyTorch model for MNIST

About PyTorch model training

Perform training and save the model

Use the model for inference

Understand inference on Android

Create an Android application

Prepare Test Data

Run the Application

Optimizing neural network models in PyTorch

Create an optimized PyTorch model for MNIST

Run optimization

Update the Android application

Review

Next Steps

Feedback

How would you rate the overall quality of this Learning Path?

What is the primary reason for your feedback ?

Missing information

Too complicated / unclear steps

Code problems

Out of date

Easy to understand

Solved my problem

Excellent code snippets

Please select a reason before submitting.

Thank you. We're grateful for your feedback on how to improve this Learning Path.

Have more detailed feedback about what to improve or add next? Log an issue on GitHub.
Want to discuss this topic with fellow developers and Arm Experts? Collaborate in the Arm Developer Discord server.

Continue Learning

Proceed to Use Keras Core with TensorFlow, PyTorch, and JAX backends to continue exploring Machine Learning.

Next Learning Path: Use Keras Core with TensorFlow, PyTorch, and JAX backends

Similar Learning Paths

Learn about related topics:

Build a Windows on Arm native application with clang
Get started with Scalable Vector Extension 2 (SVE2) on Android

Explore Tags

Find additional Learning Paths using tags:

Cortex-A

Neoverse

Windows

Linux

macOS

Android Studio

Coding

VS Code

Back

Next Steps

Create and train a PyTorch model for digit classification

Introduction

Prepare a PyTorch development environment

Create a PyTorch model for MNIST

About PyTorch model training

Perform training and save the model

Use the model for inference

Understand inference on Android

Create an Android application

Prepare Test Data

Run the Application

Optimizing neural network models in PyTorch

Create an optimized PyTorch model for MNIST

Run optimization

Update the Android application

Review

Next Steps

Create and train a PyTorch model for digit classification

Feedback

Continue Learning

Similar Learning Paths

Further Reading

Explore Tags