Pytorch中torch.load()中出现AttributeError: Can't get attribute
原因:保存下来的模型和参数不能在没有类定义时直接使用。
Pytorch使用Pickle来处理保存/加载模型,这个问题实际上是Pickle的问题,而不是Pytorch。
解决方法也非常简单,只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下,再import Class!
但是,在实际过程中,我们采取import Class的方法没起到作用,而直接在要运行的文件上复制类定义是一种可行的方法。
本文介绍了解决PyTorch中使用torch.load()时遇到的AttributeError问题,该问题源于Pickle处理保存/加载模型时无法找到类定义。文章提供了具体的解决方法,即确保类定义可见。
4310

被折叠的 条评论
为什么被折叠?



