Making Predictions with PyTorch Deep Learning Models
Learn how to load PyTorch models and to make inferences. As a toy use case, we’ll work with a classifier to identify different digits in the MNIST dataset.
Code:
Making predictions
===============================
Interested in hiring me as a consultant/freelancer?
Join The Sound Of AI Slack community:
Connect with Valerio on Linkedin:
Follow Valerio on Facebook:
Follow Valerio on Twitter:
===============================
Content:
0:00 Intro
1:08 Implementation steps
1:54 Loading PyTorch model
3:27 Loading MNIST dataset
4:20 Getting a sample from dataset
5:42 Making an inference
6:37 Class mapping
8:42 Predict function
14:50 Running the