pranerd / MNIST_Pytorch_python_and_capi

This is an example of how to train a MNIST network in Python and run it in c++ with pytorch 1.0

MNIST_Pytorch_python_and_capi

1. Run main of pythontracing.py --> training the MNIST network and saving it to model.pth
2. Run demo of pythontracing.py --> tracing the model.pth so it can be used in c++
3. Build the example-app.cpp using CMake
4. Run it!