- Run main of pythontracing.py --> training the MNIST network and saving it to model.pth
- Run demo of pythontracing.py --> tracing the model.pth so it can be used in c++
- Build the example-app.cpp using CMake
- Run it!
-
Notifications
You must be signed in to change notification settings - Fork 8
tobiascz/MNIST_Pytorch_python_and_capi
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
This is an example of how to train a MNIST network in Python and run it in c++ with pytorch 1.0
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published