pytorch_deep_learning - HxGN EAM - 12.2 - Feature Briefs - Hexagon

HxGN EAM Python Framework Technical Reference

ft:locale
en-US
Product
HxGN EAM
Search by Category
Feature Briefs
HxGN EAM Version
12.2

This example illustrates PyTorch 1.12.0 deep learning classification procedures using built-in MNIST data set. This is also a good exercise of PyTorch framework learning. From data generation, batch Tensor data loader creation/usage, dense neural network MLP definition and model definition with loss function and optimizer definition, model training and evaluation, model saving /loading and predict comparison, result visualization, etc.