分类：grad

2020
10-08

pytorch查看模型weight与grad方式

在用pdbdebug的时候，有时候需要看一下特定layer的权重以及相应的梯度信息，如何查看呢？1.首先把你的模型打印出来，像这样2.然后观察到model下面有module的key，module下面有features的key，features下面有(0)的key，这样就可以直接打印出weight了，在pdbdebug界面输入pmodel.module.features[0].weight，就可以看到weight，输入pmodel.module.features[0].weight.grad就可以查看梯度信息补充知识：查看Pytorch网络的各层... 继续阅读 >