README.md

Mnist PyTorch example

This example has branches to show case CPU based and GPU based (amd, nvidia) local training of machine learning application.

Refer following for example execution.

CPU based training

GPU based with AMD GPUs

GPU based with Nvidia GPUs