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

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

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

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

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

猜你喜欢

转载自www.cnblogs.com/lucifer1997/p/11412407.html