[深度學習]pytorch部署之onnx

最近嘗試使用onnx來部署torch模型，發現還是有一些坑的： 1、盡量使用經典模型結構，模型的輸入不要增加內容（比如bert的輸入增加一個label_ids），這會導致onnx模型的輸入無法識別l ...