pytorch 如何加载部分预训练模型
2017-03-19 14:55
661 查看
pretrained_dict =...
model_dict = model.state_dict()
# 1. filter out unnecessary keys
pretrained_dict = {k: v for k, vin pretrained_dict.items() if k inmodel_dict}
# 2. overwrite entries in the existing state dict
model_dict.update(pretrained_dict)
# 3. load the new state dict
model.load_state_dict(model_dict)
model_dict = model.state_dict()
# 1. filter out unnecessary keys
pretrained_dict = {k: v for k, vin pretrained_dict.items() if k inmodel_dict}
# 2. overwrite entries in the existing state dict
model_dict.update(pretrained_dict)
# 3. load the new state dict
model.load_state_dict(model_dict)
相关文章推荐
- PyTorch(7)——模型的训练和测试、保存和加载
- 如何用Tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件
- 如何用Tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件
- pytorch加载已训练好的word-embedding
- PyTorch学习系列(十五)——如何加载预训练模型?
- 如何用Tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件
- PyTorch中使用预训练的模型初始化网络的一部分参数
- pytorch多GPU训练以及多线程加载数据
- 【深度学习】笔记7: CNN训练Cifar-10技巧 ---如何进行实验,如何进行构建自己的网络模型,提高精度
- 如何实现ZBrush中部分模型的选择和隐藏
- windows下用caffe加载二进制模型(linux下训练)的问题
- Learning Caffe(9)---加载训练好的模型进行识别
- 如何用kaldi训练好的DNN模型做在线识别
- TensorFlow保存和加载训练模型
- 4000 TensorFlow学习笔记(2)——保存和加载训练模型参数
- tensorflow学习笔记六:保存和加载训练模型
- tsnsorflow学习笔记七:详解如何将TensorFlow训练的模型移植到Android手机
- Torch 7 load saved model failed, 加载保存的模型失败
- c++ 调用py-faster-rcnn训练好的模型
- (原)ubuntu16在torch中使用caffe训练好的模型