修改训练好的神经网络模型中的参数内容

在迁移学习中，将训练好的模型用在其他地方上，当需要对模型的参数进行修改，可按照如下代码进行操作：

path = "/media/FASTSCNN/lanenet_epoch_227_batch_8.model"  #训练好的模型

state_dict2 = {"module."+k: v for k, v in torch.load(path).items()}  
#此处是在原本模型的参数名的基础上，多加"module."的字符串。即如果原参数名为“conv1”，修改后的参数名为”module.conv1“

path = "/media/FASTSCNN/lanenet_epoch_227_batch_8_1.model"  #修改后保存的地址
torch.save(state_dict2, path)   #模型保存