將模型從PYTORCH導出到ONNX並使用ONNX RUNTIME運行

在本教程中,我們描述瞭如何將PyTorch中定義的模型轉換爲ONNX格式,然後在ONNX Runtime中運行它。 ONNX Runtime是針對ONNX模型的以性能爲中心的引擎,可在多個平臺和硬件(Windows,Linux和Mac以及CPU和GPU上)高效地進行推理。ONNX運行時已被證明大大增加了多種型號的性能,說明這裏 對於本教程,大家將需要安裝ONNX 和ONNX Runtime。大家可
相關文章
相關標籤/搜索