分类：模型复制

2021
03-05

解决pytorch 模型复制的一些问题

直接使用model2=model1会出现当更新model2时，model1的权重也会更新，这和自己的初始目的不同。经评论指出可以使用：model2=copy.deepcopy(model1)来实现深拷贝，手上没有pytorch环境，具体还没测试过，谁测试过可以和我说下有没有用。原方法：所有要使用模型复制可以使用如下方法。torch.save(model,"net_params.pkl")model5=Cnn(3,10)model5=torch.load('net_params.pkl')这样编写不会影响原始模型的权重补充：pytorch模型训练... 继续阅读 >