A colab_nngen_lenet.ipynb
file contains steps needed to:
- Setup environment and install NNgen in Google Colab service,
- Load a neural network model saved in ONNX format and implemented using Torch. As we tested, models saved with Tensorflow do not work with NNgen,
- Generate IP block for Vivado,
- Quantize and save model weights.