- 保存模型
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')
- 加载模型
net = torch.load('resourses/net_trained.pkl')
此处的net为新的实例化对象