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

作者：2023面试高手 | 2024-03-10 08:00:59

踩

已保存的神经网络改参数

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

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)   #模型保存
1
2
3
4
5
6
7

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/217563