Support for pytorch? - Githubissues

ARM-software / armnn

Arm NN ML Software. The code here is a read-only mirror of https://review.mlplatform.org/admin/repos/ml/armnn

https://developer.arm.com/products/processors/machine-learning/arm-nn

MIT License

1.14k stars 307 forks source link

Support for pytorch? #713

Closed EilersJan closed 1 year ago

EilersJan commented 1 year ago

Hey guys, are there any news to executing pytorch models with ArmNN?

On pytorch mobile they only mention XNNPack and QNNPack..

MikeJKelly commented 1 year ago

Hi @EilersJan

At the moment we have no plans to support pytorch from within ArmNN. At the moment we are evaluating the use of TinyNeuralNetwork to convert from PyTorch to a TfLite model. Maybe you can give that a go and let us know how it works for you?

https://github.com/alibaba/TinyNeuralNetwork

Best regards, Mike

EilersJan commented 1 year ago

Thanks Mike, was hoping you might be working on a delegate for pytorch mobile