pytorch中保存模型
在pytorch中有两种模型保存方式,如下所示:
# 第一种:只保留模型参数:
torch.save(model.state_dict(),path) # 保存
model.load_state_dict(torch.load(path)) # 加载(需提前定义模型)
# 第二种:保存整个模型:
torch.save(model,path) # 保存模型
model = torch.load(path) # 加载模型
- 上一篇 »PyTorch模型加载与保存的最佳实践
- 下一篇 »pytorch两种模型保存方式