【深度学习torch——error】——“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)
生活随笔
收集整理的這篇文章主要介紹了
【深度学习torch——error】——“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
這個問題是在進行權重文件加載進行預測的時候發生的,原因其實就是torch版本不對
我是用的工作站訓練使用的是torch1.7.0,然后用自己的電腦進行預測,就報錯了,原因就是自己的電腦是torch1.2.0版本的
因為在1.6版本以上的模型改變了權重文件的讀取格式,所以盡量版本之間不要差得太大
總結
以上是生活随笔為你收集整理的【深度学习torch——error】——“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: BZOJ4293 Siano
- 下一篇: adb重启方法