PyTorch学习笔记（4）--神经网络模型的保存和导入

1.numpy矩阵的保存

import numpy as np
a=np.array(2)
np.save("nm.npy",a)
a=np.load("nm.npy")

2.模型的保存和导入

将训练好的模型和参数保存下来，下一次使用的时候直接导入模型和参数，和一个已经训练好的神经网络模型一样

保存模型

import torch
#保存整个神经网络的结构和模型参数
torch.save(mymodel,'mymodel.pkl')
#只保存整个神经网络的模型参数
torch.save(mymodel.state_dict(),"mymodel_params.pkl")

导入模型

mymodel=torch.load('mymodel.pkl')

参考书籍：《PyTorch深度学习实战（侯宜君著）》