程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式師世界 >> 編程語言 >  >> 更多編程語言 >> Python

通過c++加載python訓練好的模型,libtorch

編輯:Python
問題遇到的現象和發生背景

通過c++加載python訓練好的模型,libtorch

問題相關代碼,請勿粘貼截圖
torch::jit::script::Module model_;model_ = torch::jit::load(model);
運行結果及報錯內容

terminate called after throwing an instance of 'c10::Error'
what(): PytorchStreamReader failed locating file constants.pkl: file not found
Exception raised from valid at ../caffe2/serialize/inline_container.cc:158 (most recent call first):
frame #0: c10::Error::Error(c10::SourceLocation, std::__cxx11::basic_string ) + 0x6b (0x7f5bee69175b in ../lib/libtorch/libc10.so)
frame #1: c10::detail::torchCheckFail(char const*, char const*, unsigned int, std::__cxx11::basic_string const&) + 0xce (0x7f5bee68d38e in ../lib/libtorch/libc10.so)
frame #2: caffe2::serialize::PyTorchStreamReader::valid(char const*, char const*) + 0x3ca (0x7f5bdacea09a in ../lib/libtorch/libtorch_cpu.so)

我想要達到的結果

正確加載


  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式師世界 All Rights Reserved