pytroch调试时显示网络中的变量大小或具体数值
首先设置断点,打开调试窗口。在这里插入代码片
这个时候想看各个变量的具体值是多少,直接输入conv1.parameter是没有显示的:
需要这样:
for name, para in net.conv1.named_parameters():
print(name, para.size())
查看某一层的权重:
print(net.conv1.weight.data)
查看某一层偏置的梯度:
print(net.conv1.bias.grad)