Bootstrap

object has no attribute ‘...’ in torch.load

torch.load 加载预训练模型的参数,可能会遇到下面报错

AttributeError: ‘...’ object has no attribute ‘copy’
AttributeError: ‘...’ object has no attribute ‘module’

引发的原因可能是:

使用了torch.save(model)保存模型参数,但使用model.torch.load()加载模型

正确的做法:

使用了torch.save(model)保存模型参数,加载时model = torch.load('.pth')
使用了torch.save(model.state_dict())保存模型参数,加载时model.torch.load()

参考文档

;