花费 5 ms
[深度学习]pytorch部署之onnx

最近尝试使用onnx来部署torch模型,发现还是有一些坑的: 1、尽量使用经典模型结构,模型的输入不要增加内容(比如bert的输入增加一个label_ids),这会导致onnx模型的输入无法识别l ...

Tue Jul 13 23:36:00 CST 2021 0 171

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM