如何解決PyTorch中的Nomodulenamedmodels

如何解決PyTorch中的No module named models，相信很多沒有經(jīng)驗的人對此束手無策，為此本文總結(jié)了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個問題。

創(chuàng)新互聯(lián)建站成立于2013年，是專業(yè)互聯(lián)網(wǎng)技術(shù)服務(wù)公司，擁有項目成都網(wǎng)站設(shè)計、網(wǎng)站制作網(wǎng)站策劃，項目實施與項目整合能力。我們以讓每一個夢想脫穎而出為使命，1280元洱源做網(wǎng)站,已為上家服務(wù),為洱源各地企業(yè)和個人服務(wù),聯(lián)系電話:13518219792

軟硬件環(huán)境

  • ubuntu 18.04 64bit
  • anaconda with 3.7
  • nvidia gtx 1070Ti
  • cuda 10.1
  • pytorch 1.5
 

問題

在使用pytorch深度學習框架訓練出來的模型文件，在另外的工程中使用，經(jīng)常會碰到如下的錯誤

  File "/home/xugaoxiang/anaconda3/envs/torchTest/lib/python3.7/site-packages/torch/serialization.py", line 593, in load    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)  File "/home/xugaoxiang/anaconda3/envs/torchTest/lib/python3.7/site-packages/torch/serialization.py", line 773, in _legacy_load    result = unpickler.load()ModuleNotFoundError: No module named 'models'
 
 

解決方法

其實這個問題，在pytorch的官方文檔中就有提到，如下

如何解決PyTorch中的No module named models

上面這種方法呢是推薦的做法，在執(zhí)行torch.save和torch.load時，操作的都是模型的參數(shù)，這樣移植起來非常的方便

而下面的方法則是針對整個模型，在訓練模型的時候，會將本地的class和目錄結(jié)構(gòu)都寫入到模型中。很多開源項目在模型訓練完成后也是采用這樣的方法來保存，因此，當在你自己的項目中去使用這樣的模型時，往往就會遇到上面出現(xiàn)的問題，解決的方法就是在你的項目中保持原有項目的必要結(jié)構(gòu)，如相應(yīng)的class和模塊

如何解決PyTorch中的No module named models

看完上述內(nèi)容，你們掌握如何解決PyTorch中的No module named models的方法了嗎？如果還想學到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝各位的閱讀！

名稱欄目：如何解決PyTorch中的Nomodulenamedmodels
分享網(wǎng)址：http://muchs.cn/article42/iehjec.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供動態(tài)網(wǎng)站、標簽優(yōu)化、定制網(wǎng)站、企業(yè)建站、搜索引擎優(yōu)化、網(wǎng)站設(shè)計公司

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)

h5響應(yīng)式網(wǎng)站建設(shè)