PyTorch錯誤解決:XXX is a zip archive(did you mean to use torch.jit.load()?)


錯誤原因:

訓練保存模型時,torch的版本是1.6.0(使用torch.__version__可以查看torch的版本號

而加載模型時,torch的版本號低於1.6.0

 

解決方案:

If for any reason you want torch.save to use the old format, pass the kwarg _use_new_zipfile_serialization=False.

在訓練時更改參數“_use_new_zipfile_serialization=False”,

torch.save(model.state_dict(), model_cp,_use_new_zipfile_serialization=False)  # 訓練所有數據后,保存網絡的參數

 

參考:https://blog.csdn.net/weixin_44769214/article/details/108188126

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM