PyTorch模型持久化（模型的保存和加载）

#PyTorch保存整个模型和保存模型的参数
torch.save(mlpl,"data/mlpl.pkl")#保存整个模型
mlplload=torch.load("data/mlpl.pkl")#导入保存的模型
print(mlplload)
#只保存模型的参数 mlpl.state_dict()获取网络中已经训练好的参数
torch.save(mlpl.state_dict(),"data/mlpl_params.pkl")
mlpl_params=torch.load("data/mlpl_params.pkl")
print(mlpl_params)