[Pytorch] 保存模型与加载模型

1、保存模型

# 定义模型
model = BPNetModel(n_feature=n_feature,n_hidden=n_hidden,n_output=n_output) #调用网络

# 保存模型
torch.save(model, 'BPNetModel0.pth')

2、加载模型

import torch

## 读取模型
model = torch.load('BPNetModel0.pth')

3、保存模型参数

 #调用网络
model = BPNetModel(n_feature=n_feature,n_hidden=n_hidden,n_output=n_output)

# 保存模型
torch.save({'model': model.state_dict()}, 'BPNetModel0.pth')

4、加载参数

# 读取模型
state_dict = torch.load('model_name.pth')
model.load_state_dict(state_dict['model'])