溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
如何解決PyTorch中的No module named models，相信很多沒有經驗的人對此束手無策，為此本文總結了問題出現的原因和解決方法，通過這篇文章希望你能解決這個問題。
在使用pytorch深度學習框架訓練出來的模型文件，在另外的工程中使用，經常會碰到如下的錯誤
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的官方文檔中就有提到，如下
上面這種方法呢是推薦的做法，在執行torch.save和torch.load時，操作的都是模型的參數，這樣移植起來非常的方便
而下面的方法則是針對整個模型，在訓練模型的時候，會將本地的class和目錄結構都寫入到模型中。很多開源項目在模型訓練完成后也是采用這樣的方法來保存，因此，當在你自己的項目中去使用這樣的模型時，往往就會遇到上面出現的問題，解決的方法就是在你的項目中保持原有項目的必要結構，如相應的class和模塊
看完上述內容，你們掌握如何解決PyTorch中的No module named models的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
