Pytorch保存和加载训练好的模型，以及加载中提示Can‘t get attribute ‘XXX‘ on ＜module ‘__main__‘ from ‘YYY‘＞的解决方法

保存，分仅保存参数，和保存整个模型

1 仅保存参数，轻便，但需把网络模型先执行一遍

方法百度

2 保存整个模型，无脑全存

保存

torch.save(rnn, R'D:\Python\11111.pkl')

加载.

rnn = torch.load(R'D:\Python\11111.pkl')

如果直接执行，报错

AttributeError: Can't get attribute 'RNN' on <module '__main__' from '<input>'>

解决方法：

把模型先加载一遍，无需保证超参数完全一致，也无需训练