pytorch查看模型weight與grad

在用pdb debug的時候，有時候需要看一下特定layer的權重以及相應的梯度信息，如何查看呢？ 1. 首先把你的模型打印出來，像這樣 2. 然後觀察到model下面有module的key，module下面有features的key， features下面有(0)的key，這樣就可以直接打印出weight了，在pdb debug界面輸入p model.module.features[0].wei

>>阅读原文<<