pytorch保存最终模型方式：①、直接保存模型，加载模型的时候直接把读取的模型给一个参数就行；②、只保存参数，在读取模型参数前要先定义一个模型（模型必须与原模型相同的构造），然后对这个模型导入参数

一、直接保存整个模型并读取

第一种方法可以直接保存模型，加载模型的时候直接把读取的模型给一个参数就行。

# 创建你的模型实例对象: model
model = net()

## 保存模型
torch.save(model, 'model_name.pth')

## 读取模型
model = torch.load