机器学习-网络模型的保存与使用（pytorch环境）

例子

vgg16_false = torchvision.models.vgg16(pretrained=False)

# 保存方式1（网络模型）
torch.save(vgg16_false,"vgg_method1.pth")

# 保存方式2（参数模型）
torch.save(vgg16_false.load_state_dict(),"vgg_method2.pth")

# 陷阱1：如果要保存的模型中有自定义的类，则需要在加载时导入，否则将会报错

# 加载方式1
model1 = torch.load("vgg_method1.pth")
print(model1)

# 加载方式2（官方推荐）（将参数模型转换为网络模型）
model2 = torchvision.models.vgg16(pretrained=False)
model2.load_state_dict("vgg_method2.pth")
print(model2)

其它待补。。。