Pytorch查看tensor是否有梯度（值）以及开启梯度-Toy模板网

这篇具有很好参考价值的文章主要介绍了Pytorch查看tensor是否有梯度（值）以及开启梯度。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

一. requires_grad 属性：查看是否记录梯度

x = torch.rand(3, 3)   # 直接创建的tensor变量默认是没有梯度的
x.requires_grad

结果如下：

False

二. requires_grad_ ()函数：调用函数设置记录梯度与否

函数：requires_grad_(requires_grad=True)

x = torch.tensor([1., 2., 3.])
print('x_base: ', x.requires_grad)   # 正常状态：False
x.requires_grad_(False) #False
print('x.requires_grad_(False): ', x.requires_grad) #False
x.requires_grad_(True)  # 等价：requires_grad_() #True
print('x.requires_grad_(True): ', x.requires_grad) #True

结果如下：

x_base:  False
x.requires_grad_(False):  False
x.requires_grad_(True):  True

三. requires_grad属性参数，创建tensor时设置是否记录梯度

x = torch.tensor([1., 2., 3.], requires_grad=True) #True
print('x_base: ', x.requires_grad)   # 正常状态

结果如下：

x_base:  True

四. 查看模型的权重名称和参数值

[name for name, param in model.named_parameters()] #查看模型的权重名称
[param for name, param in model.named_parameters()] #查看模型的权重值

五. 查看模型权重梯度值

网络结构如下：
Pytorch查看tensor是否有梯度（值）以及开启梯度

class Simple(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(3, 16, 3, 1, padding=1, bias=False)
        self.conv2 = nn.Conv2d(16, 32, 3, 1, padding=1, bias=False)
        self.linear = nn.Linear(32*10*10, 20, bias=False)
 
    def forward(self, x):
        x = self.conv1(x)
        x = self.conv2(x)
        x = self.linear(x.view(x.size(0), -1))
        return x
 
model = Simple()