#PyTorch保存整个模型和保存模型的参数
torch.save(mlpl,"data/mlpl.pkl")#保存整个模型
mlplload=torch.load("data/mlpl.pkl")#导入保存的模型
print(mlplload)
#只保存模型的参数 mlpl.state_dict()获取网络中已经训练好的参数
torch.save(mlpl.state_dict<
PyTorch模型持久化(模型的保存和加载)
最新推荐文章于 2023-06-26 14:04:18 发布