Pytorch模型保存与加载

最新推荐文章于 2024-09-18 21:47:51 发布

qq_23976207

最新推荐文章于 2024-09-18 21:47:51 发布

阅读量1k

点赞数

文章标签： pytorch python

本文链接：https://blog.youkuaiyun.com/qq_23976207/article/details/122577912

版权

#模型保存与加载
#方式1
vgg16 = torchvision.models.vgg16(pretrained=False)
torch.save(vgg16,"saved_vgg16_1.pth")

#方式2
torch.save(vgg16.state_dict(),"saved_vgg16_2.pth")

#加载
#方式1
vgg16 = torch.load("saved_vgg16_1.pth")
print("方式1:",vgg16)
#方式2

vgg16 = torchvision.models.vgg16(pretrained=False)
vgg16.load_state_dict(torch.load("saved_vgg16_2.pth"))
print("方式2:",vgg16)