当前位置：网站首页>pytorch：保存和加载模型

pytorch：保存和加载模型

2022-08-02 03:25:00 【PenguinLeee】

之前跑了好几个模型，都需要保存和加载模型信息。为了省得每次都上网找，开一个帖记录一下。

此处只保存模型参数

model = torch.nn.Linear(1, 2)
torch.save(model.state_dict(), "./model.pth")

model = torch.nn.Linear(1, 2)
model.load_state_dict(torch.load("./model.pth"))

版权声明
本文为[PenguinLeee]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_43466027/article/details/121363300