【TVM 學(xué)習(xí)資料】編譯 PaddlePaddle 模型
本篇文章譯自英文文檔?Compile PaddlePaddle Models (https://tvm.apache.org/docs/how_to/compile_models/from_paddle.html#sphx-glr-download-how-to-compile-models-from-paddle-py)
作者是?Ziyuan Ma
(https://github.com/ZiyuanMa/)
更多 TVM 中文文檔可訪問 →Apache TVM 是一個(gè)端到端的深度學(xué)習(xí)編譯框架,適用于 CPU、GPU 和各種機(jī)器學(xué)習(xí)加速芯片。 | Apache TVM 中文站
(https://tvm.hyper.ai/)
本文介紹如何用 Relay 部署 PaddlePaddle 模型,首先安裝 PaddlePaddle(版本>=2.1.3),可通過 pip 快速安裝:
或參考官方網(wǎng)站:https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
輸出結(jié)果:
加載預(yù)訓(xùn)練的 ResNet50 模型
加載 PaddlePaddle 提供的 ResNet50 預(yù)訓(xùn)練模型:
輸出結(jié)果:
加載測(cè)試圖像
還是用貓的圖像:
使用 Relay 編譯模型
輸出結(jié)果:
在 TVM 上執(zhí)行
查找分類集名稱
在 1000 個(gè)類的分類集中,查找分?jǐn)?shù)最高的第一個(gè):
輸出結(jié)果:
下載 Python 源代碼:from_paddle.py
(https://tvm.apache.org/docs/_downloads/16269b77359771348d507395692524cf/from_paddle.py)
下載 Jupyter Notebook:from_paddle.ipynb
(https://tvm.apache.org/docs/_downloads/a608d8b69371e9bc149dd89f6db2c38e/from_paddle.ipynb)