极术小姐姐 · 2021年09月29日

用于在Arm Ethos-U55和U65上部署的PyTorch到TensorFlow Lite

image.png

目前,如果你想在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技术专栏

推荐阅读
关注数
23387
内容数
893
Arm相关的技术博客,提供最新Arm技术干货,欢迎关注
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息