pytorch加载已经保存的模型文件，并将其作为另一个网络的预训练权重

---

问题描述

（自己做笔记用）之前将训练好的模型保存起来，现在想把模型加载进来。

解决办法

#加载保存好的模型
Layer1pre = torch.load('./ResultData_earlystop/savemodel/checkpoint_model_layer1.pt')

#定义自己的模型
model = CNNLayer(num_classes=10, aux_logits=True)
if use_gpu:
    model = model.cuda()

#将模型权重更新到新的网络中
model.load_state_dict(Layer1pre, strict=False)

参考链接：https://blog.csdn.net/my_kingdom/article/details/85218478