Pytorch中torch.load()中出现AttributeError: Can't get attribute


原因:保存下来的模型和参数不能在没有类定义时直接使用。

Pytorch使用Pickle来处理保存/加载模型,这个问题实际上是Pickle的问题,而不是Pytorch。

解决方法也非常简单,只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下,再import Class!

但是,在实际过程中,我们采取import Class的方法没起到作用,而直接在要运行的文件上复制类定义是一种可行的方法。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM