pytorch保存模型和导入模型

参考

model.state_dict()中保存了{参数名:参数值}的字典

import torchvision.models as models

resnet34 = models.resnet34(pretrained=True)
resnet34.state_dict().keys()

保存模型
torch.save(model.state_dict(), PATH) # 保存模型为pth

导入模型

model = ModelClass()   # 需要先建立模型
model.load_state_dict(torch.load(PATH)) # 加载模型

pytorch保存模型和导入模型

原文：https://www.cnblogs.com/huanxifan/p/12656926.html