Dali杂货铺🐰17——torch 训练过程查看某一层的梯度

"Drawing neural network"

Posted by fuhao7i on April 5, 2021

loss.backward()之后查看某一层的梯度

1
2
print('=== after loss.backward() ===')
print(self.model.conv0.weight.grad)