Create and train a PyTorch model for digit classification using the MNIST dataset
Introduction
Prepare a PyTorch Development Environment
Create a PyTorch model for MNIST
About PyTorch Model Training
Perform Training and Save the Model
Deploy the Model for Inference
Learn about Inference on Android
Create an Android Application
Prepare the Test Data
Run the Application
Optimizing Neural Network Models in PyTorch
Create an optimized PyTorch model for MNIST
Run optimization
Update the Android application
Next Steps

  Log an issue

  Fork and edit

  Discuss on Discord

Share

Bring your insights to the conversation.

Give Feedback

How would you rate this Learning Path?

Continue Learning

Read related resources

Find more information about the topics in this Learning Path:

Join the Arm Developer Program

Connect, upskill, and build with the Arm Developer Community. Join today for hands-on technical resources and education materials, along with the support of Arm engineers and the broader ecosystem.

Join the Arm Developer Program

By signing up, you will receive member news and email updates on topics that interest you. You can manage your subscription preferences anytime.

Developer Program

This data is not linked your email address or name. It is aggregated and reported internally as part of Arm's DEI program By signing up, you'll receive member news and email updates. You can update your subscription preferences at any time.

Arm will process your information in accordance with our Privacy Policy.

Arm will process your information in accordance with our Arm and Arm China Privacy Policies.

I consent to have my data transferred outside of China.

Submit Close
Join now

Learn more on Arm Developer

Visit Developer.arm.com to continue your learning journey.

Developer.arm.com
Back