18.5使用tensorrt加速tensorflow的預測/前向傳播速度

本文接着前面的18.1至18.3博客。html nvidia推出的tensorrt能夠加速前向傳播的速度。本文采用tensorflow訓練好的mobilenetv2模型進行測試,按照前面博客咱們已經能夠把訓練好的模型轉爲.pb的格式了,這裏的tensorrt就是對.pb文件進行加速。python 1、tensorrt安裝git 安裝方法參考https://developer.download.nv
相關文章
相關標籤/搜索