pytorch框架下两种模型保存及加载方式
方法1
以字典(key-value)保存模型的参数
torch.save(vgg16.state_dict(), "路径")
vgg16.load_state_dict(torch.load("路径"))
方法2
保存模型参数及模型结构
torch.save(vgg16, "路径h")
model = torch.load("路径")
pytorch框架下两种模型保存及加载方式
方法1
以字典(key-value)保存模型的参数
torch.save(vgg16.state_dict(), "路径")
vgg16.load_state_dict(torch.load("路径"))
方法2
保存模型参数及模型结构
torch.save(vgg16, "路径h")
model = torch.load("路径")