目前,如果你想在Arm Ethos-U55或Arm Ethos-U65上部署你的机器学习模型,你需要使用TensorFlow Lite for Microcontrollers。这样做需要你首先将你的模型以TensorFlow Lite的格式存在。如果你已经用TensorFlow训练了你的模型,那么转换为TensorFlow Lite的过程已经有很好的记录了。
然而,你可能已经用PyTorch训练了你的模型,而PyTorch本身并不允许你输出到TensorFlow Lite格式。本指南将展示如何使用ONNX将你训练的PyTorch模型转换为TensorFlow Lite,然后对其进行量化,以便在Arm Ethos-U55或Arm Ethos-U65上部署。
阅读直达链接:https://community.arm.com/developer/ip-products/processors/b/ai-and-ml-blog/posts/pytorch-to-tensorflow-lite-for-deploying-on-arm-ethos-u55-and-u65
作者:Richard Burton欢迎大家点赞留言,更多Arm技术文章动态请关注极术社区Arm技术专栏