Pytorch中保存模型

pytorch中模型的保存pth,pkl,pt格式

pytorch保存模型非常简单，主要有两种方法：

1. 只保存参数；（官方推荐）
2. 保存整个模型 (结构+参数)。
   由于保存整个模型将耗费大量的存储，故官方推荐只保存参数，然后在建好模型的基础上加载。本文介绍两种方法，但只就第一种方法进行举例详解。

模型保存与调用方式一：

保存：

torch.save(model, mymodel.pth)#保存整个model的状态

调用：

model=torch.load(mymodel.pth)#这里已经不需要重构模型结构了，直接load就可以
model.eval()