最近學習基於pytorch框架下的行為識別演算法(eco**),要將訓練好的模型在c++工程中用起來,需要走通c++呼叫pytorch的模型的路。在參考了官網的說明和網上已有的部落格,記錄下遇到的問題。
1.網上別人已經寫好的**github和教程
2.官網的介面說明
遇到的問題:
cmake時需要指明libtorch的路徑且需要是絕對路徑(在windows下,需要加上-a x64來指定64位,並且要在release下才能正確的匯入模型)
cmake -dcmake_prefix_path=path/to/libtorch ..
gcc編譯器版本需要5.3.1不然會報錯:
terminate called after throwing an instance of 'std::bad_alloc'
what(): std::bad_alloc
aborted
pytorch的模型解析
如何獲取pytorch的動態圖?model torch.jit.load test.pth graph model.graph.copy torch.c.jit pass inline graph node list graph.nodes 載入模型後,獲取模型的graph,這個graph就是需要的...
安裝pytorch遇到的問題
git clone recursive pip install r requirements.txt 這步操作超級慢 python setup.py install 安裝成功後,測試 python import torch 出現錯誤 參考 cd usr local lib python3.5 dis...
pytorch 學習遇到的 問題
relu 和 leaky relu 中 inplace 設定 nn.relu inplace true default inplace false nn.leakyrelu inplace true default inplace false 將計算得到的值直接覆蓋之前的值 有時能夠節省運算記憶體,...