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
Back to Top