分类：梯度grad

2021
08-08

pytorch 权重weight 与 梯度grad 可视化操作

pytorch权重weight与梯度grad可视化查看特定layer的权重以及相应的梯度信息打印模型观察到model下面有module的key，module下面有features的key，features下面有(0)的key，这样就可以直接打印出weight了在pdbdebug界面输入pmodel.module.features[0].weight，就可以看到weight，输入pmodel.module.features[0].weight.grad就可以查看梯度信息。中间变量的梯度:.register_hookpytorch为了节省显存，在反向传播的过程中... 继续阅读 >