pytorch_1.12.0_deep_learning - HxGN EAM - Feature Briefs - Hexagon

HxGN EAM Python Studio (Flex Python) Technical Reference

Language
English
Product
HxGN EAM
Search by Category
Feature Briefs
HxGN EAM Version
12.0.0.1

This example illustrates PyTorch 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.