【深度学习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()?)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
這個(gè)問題是在進(jìn)行權(quán)重文件加載進(jìn)行預(yù)測的時(shí)候發(fā)生的,原因其實(shí)就是torch版本不對
我是用的工作站訓(xùn)練使用的是torch1.7.0,然后用自己的電腦進(jìn)行預(yù)測,就報(bào)錯(cuò)了,原因就是自己的電腦是torch1.2.0版本的
因?yàn)樵?.6版本以上的模型改變了權(quán)重文件的讀取格式,所以盡量版本之間不要差得太大
總結(jié)
以上是生活随笔為你收集整理的【深度学习torch——error】——“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: BZOJ4293 Siano
- 下一篇: 【深度学习】——模型评估指标MAP计算实