Pytorch模型保存与加载

#模型保存与加载
#方式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)