module类内部state_dict返回对象只是一个copy,所以修改里面的值并不能影响模型中真正的参数,我们可以借助tensor.copy_()方法来赋值。
简单例子:
a = torch.tensor([[1,2], [3,4]])
b = torch.tensor([[7,8],[9,10]])
a.copy_(b)
print(a)
print(b)
module类内部state_dict返回对象只是一个copy,所以修改里面的值并不能影响模型中真正的参数,我们可以借助tensor.copy_()方法来赋值。
简单例子:
a = torch.tensor([[1,2], [3,4]])
b = torch.tensor([[7,8],[9,10]])
a.copy_(b)
print(a)
print(b)