原文：Pytorch中torch.load()中出现AttributeError: Can't get attribute
原因：保存下来的模型和参数不能在没有类定义时直接使用。 Pytorch使用Pickle来处理保存 加载模型，这个问题实际上是Pickle的问题，而不是Pytorch。 解决方法也非常简单，只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下，再import Class 但是，在实际过程中，我们采取import Class的方法没起到作用，而直接在要运行的文件上复制类定义是 ...
2019-08-26 14:38 0 890 推荐指数：
相关推荐
问题原因： load的时候内存中没有School的地址，load不知道按什么格式反序列化所以报错 解决办法： 在启动文件中导入缺失的模块 ...
torch.load()的作用：从文件加载用torch.save()保存的对象。 api： torch.load(f, map_location=None, pickle_module=<module 'pickle' from '/opt ...
我用GPU训练了一个模型，想在本机（只有CPU）加载测试时出错： 根据提示，给torch.load()添加 map_location=torch.device('cpu') 参数即可 ...
Pytorch：模型的保存与加载 torch.save()、torch.load()、torch.nn.Module.load_state_dict() Pytorch 保存和加载模型后缀：.pt 和.pth 1 torch.save() [source]保存一个序列化（serialized ...
on a CPU-only machine, please use torch.load with map_loc ...
【ipython错误】 Traceback (most recent call last): File "<string>", line 1, in <module& ...