pytorch 模型部分參數的加載

如果對預訓練模型的結構進行了一些改動，在訓練的開始前希望加載未改動部分的參數，如將resnet18的第一層卷積層conv1的輸入由3通道改爲6通道的new_conv1，將分類層fc的1000類輸出改爲2類輸出的new_fc，注意：要改一下名字與原來的不同。 導入模型 myNet=ResNet() 然後就加載模型的參數，參考pytorch 如何加載部分預訓練模型 pretrained_dict=to

>>阅读原文<<