關於 ONNX 及其用 TensorRT 7.0加載

1)Onnx簡介 ONNX 定義了一種可擴展的計算圖模型、一系列內置的運算單元(OP)和標準數據類型。每個計算流圖都定義爲由節點組成的列表,並構建有向無環圖。其中每個節點都有一個或多個輸入與輸出,每個節點稱之爲一個 OP。這至關於一種通用的計算圖,不一樣深度學習框架構建的計算圖都能轉化爲它。git 現階段 ONNX 只支持推理,因此導入的模型都須要在原框架完成訓練。github 部分框架已經內嵌了
相關文章
相關標籤/搜索