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