pytorch保存和加载训练模型

1. 保存模型

import torch
torch.save(model,path)

注：1.此处的model为实例化的model，若传进自定义的网络模型测试时会报错
2.path一定要指定实际存在的文件夹，torch不会创建文件夹。并且路径后要指明文件名称，后缀为.pkl

举个例子：

#网络模型
class Net(nn.Module):...
net = Net()

#训练后保存模型
torch.save(net,'resourses/net_trained.pkl')

2. 加载模型

net = torch.load('resourses/net_trained.pkl')

此处的net为新的实例化对象