I'm a graduate student at IUPUI. I am working with pruning of Squeezenext architecture for my thesis. So I want to deploy my model on a Cortex-M. I am using Pytorch framework. So could anyone help me in using CMSIS with Pytorch?
Thanks in Advance!!!
Hi, I am working on a converter from PyTorch to CMSIS-NN.
You can find the code at https://github.com/BCJuan/torch2cmsis
Contributions are more than welcome.
The support is really restricted right now since I have just begun the library. Hope to be able to improve it in the next months :)
Hey Juan, this is great! Can't believe this isn't provided by arm
View all questions in Machine Learning forum