一、直接保存整个模型并读取
第一种方法可以直接保存模型,加载模型的时候直接把读取的模型给一个参数就行。
# 创建你的模型实例对象: model
model = net()
## 保存模型
torch.save(model, 'model_name.pth')
## 读取模型
model = torch.load
第一种方法可以直接保存模型,加载模型的时候直接把读取的模型给一个参数就行。
# 创建你的模型实例对象: model
model = net()
## 保存模型
torch.save(model, 'model_name.pth')
## 读取模型
model = torch.load
道可道,非常道;名可名,非常名。 无名,天地之始,有名,万物之母。 故常无欲,以观其妙,常有欲,以观其徼。 此两者,同出而异名,同谓之玄,玄之又玄,众妙之门。