温柔正确的人总是难以生存,因为这世界既不温柔,也不正确。——《我的青春恋爱物语果然有问题。》
保存和加载整个模型(包括模型结构和模型参数)
#保存
torch.save(模型结构的实例化, 'model.pth')
#加载(加载前不需要实例化模型结构)
model = torch.load('model.pth')
仅保存和加载模型参数
#保存
torch.save(模型结构的实例化.state_dict(), 'params.pth')
#加载(加载前需要实例化模型结构)
模型结构的实例化.load_state_dict(torch.load('params.pth'))