Train the model

Note: Due to time constraints, we won't train the neural network in this guide.. This material is only here for completeness.

Currently, support for translating a Caffe model to CMSIS-NN functions is provided. Support for other frameworks (TensorFlow, PyTorch, etc.) will come in the future.

For this guide, the model has been pretrained and can be found here. For a full guide on how to train a model with Caffe, see this tutorial.

The trained model also needs to be converted into an lmdb database. The converted files should be in

Previous Next