迁移学习：模型的保存和载入（pytorch版）

首先说明一下pytorch载入保存pth模型文件需要注意的地方：

1.载入时需要需求模型文件只是模型参数，还是模型结构+模型参数

#1.如果pth模型文件是模型参数
model = models.resnet18()
model.load_state_dict(torch.load("model_weights.pth"))
#2.如果pth模型文件
model = torc.load("model.pth")

2.同样在保存时，也可以选择两种保存方式

#1.整个模型保存
torch.save("model.pth", PATH)
#2.保存模型的参数
model = models.vgg16(pretrained=True)
torch.save(model.state_dict(),"model_weight.pth")

3.当模型的结构与模型参数结构并不完全相同时，可选择strict=False，来导入匹配的参数

model.load_state_dict(model_dict, strict=False)