pytorch 模型部分参数的加载

如果对预训练模型的结构进行了一些改动，在训练的开始前希望加载未改动部分的参数，如将resnet18的第一层卷积层conv1的输入由3通道改为6通道的new_conv1，将分类层fc的1000类输出改为2类输出的new_fc，注意：要改一下名字与原来的不同。 导入模型 myNet=ResNet() 然后就加载模型的参数，参考pytorch 如何加载部分预训练模型 pretrained_dict=to

>>阅读原文<<