如何解決PyTorch中的No module named models，相信很多沒有經(jīng)驗(yàn)的人對(duì)此束手無策，為此本文總結(jié)了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個(gè)問題。
創(chuàng)新互聯(lián)建站成立于2013年，是專業(yè)互聯(lián)網(wǎng)技術(shù)服務(wù)公司，擁有項(xiàng)目成都網(wǎng)站設(shè)計(jì)、網(wǎng)站制作網(wǎng)站策劃，項(xiàng)目實(shí)施與項(xiàng)目整合能力。我們以讓每一個(gè)夢(mèng)想脫穎而出為使命，1280元洱源做網(wǎng)站,已為上家服務(wù),為洱源各地企業(yè)和個(gè)人服務(wù),聯(lián)系電話:13518219792
在使用pytorch深度學(xué)習(xí)框架訓(xùn)練出來的模型文件，在另外的工程中使用，經(jīng)常會(huì)碰到如下的錯(cuò)誤
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'
其實(shí)這個(gè)問題，在pytorch的官方文檔中就有提到，如下
上面這種方法呢是推薦的做法，在執(zhí)行torch.save和torch.load時(shí)，操作的都是模型的參數(shù)，這樣移植起來非常的方便
而下面的方法則是針對(duì)整個(gè)模型，在訓(xùn)練模型的時(shí)候，會(huì)將本地的class和目錄結(jié)構(gòu)都寫入到模型中。很多開源項(xiàng)目在模型訓(xùn)練完成后也是采用這樣的方法來保存，因此，當(dāng)在你自己的項(xiàng)目中去使用這樣的模型時(shí)，往往就會(huì)遇到上面出現(xiàn)的問題，解決的方法就是在你的項(xiàng)目中保持原有項(xiàng)目的必要結(jié)構(gòu)，如相應(yīng)的class和模塊
看完上述內(nèi)容，你們掌握如何解決PyTorch中的No module named models的方法了嗎？如果還想學(xué)到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝各位的閱讀！
重慶分公司
重慶分公司
