This discussion has been locked.

You can no longer post new replies to this discussion. If you have a question you can start a new discussion

Pytorch framework for Arm NN (CMSIS)

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!!!

Top replies

Karl Fezer over 4 years ago +1

Right now we're still working on support for PyTorch, especially for M class. I would recommend looking at Tensorflow Lite Micro for the time being.
Karl Fezer over 3 years ago in reply to mfber +1

Wow, sorry I missed this for so long. I would say we are looking into it, but consistently recommend testing in PyTorch and then rebuilding in Tensorflow if TFLu is the end goal.

Parents

0 Rohit Sharma over 3 years ago

Watch "Bringing PyTorch Models to Cortex-M" at ARM AI Virtual Tech Talk, on www.youtube.com/watch
Cancel
Up 0 Down

Cancel

Reply

0 Rohit Sharma over 3 years ago

Watch "Bringing PyTorch Models to Cortex-M" at ARM AI Virtual Tech Talk, on www.youtube.com/watch
Cancel
Up 0 Down

Cancel

Children

No data