MNISTPytorchpythonandcapi

October 9, 2018 ยท View on GitHub

  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!