直接使用
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')
这样编写不会影响原始模型的权重