保存和加载模型
在本节中,我们将学习如何通过保存、加载模型的状态来持久化模型,并运行模型预测。
一、 保存和加载模型权重
PyTorch 模型将学习到的参数存储在一个内部状态字典中,称为 state_dict
。
这些参数可以通过 torch.save
方法保存下来:
model = models.vgg16(weights='IMAGENET1K_V1')
torch.save(model.state_dict(
在本节中,我们将学习如何通过保存、加载模型的状态来持久化模型,并运行模型预测。
PyTorch 模型将学习到的参数存储在一个内部状态字典中,称为 state_dict
。
这些参数可以通过 torch.save
方法保存下来:
model = models.vgg16(weights='IMAGENET1K_V1')
torch.save(model.state_dict(
道可道,非常道;名可名,非常名。 无名,天地之始,有名,万物之母。 故常无欲,以观其妙,常有欲,以观其徼。 此两者,同出而异名,同谓之玄,玄之又玄,众妙之门。