Bootstrap

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

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

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

# 创建你的模型实例对象: model
model = net()
## 保存模型
torch.save(model, 'model_name.pth')

## 读取模型
model = torch.load
;