2021年7月13日
[深度学习]pytorch部署之onnx
摘要： 最近尝试使用onnx来部署torch模型，发现还是有一些坑的： 1、尽量使用经典模型结构，模型的输入不要增加内容（比如bert的输入增加一个label_ids），这会导致onnx模型的输入无法识别label_ids 解决方法：如果模型魔改又需要部署，那只能自己写导出为onnx的代码了 2、使用别的包 阅读全文
posted @ 2021-07-13 15:36 justDoIT& 阅读(773) 评论(0) 推荐(0) 编辑