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()