pytorch查看模型weight与grad

本文转载自查看原文 2019-01-30 10:58 2185 pytorch

在用pdb debug的时候，有时候需要看一下特定layer的权重以及相应的梯度信息，如何查看呢？

1. 首先把你的模型打印出来，像这样

2. 然后观察到model下面有module的key，module下面有features的key， features下面有(0)的key，这样就可以直接打印出weight了，在pdb debug界面输入p model.module.features[0].weight，就可以看到weight，输入 p model.module.features[0].weight.grad就可以查看梯度信息

免责声明！

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 Pytorch中with torch.no_grad()或@torch.no_grad() 用法 pytorch 中 torch.no_grad()、requires_grad、eval() PyTorch 中 weight decay 的设置 Pytorch打印查看模型参数总数 PyTorch笔记--torch.autograd.grad() pytorch中的 requires_grad和volatile pytorch requires_grad = True的意思 pytorch函数zero_grad(),step()作用 pytorch: grad can be implicitly created only for scalar outputs PyTorch grad_fn的作用以及RepeatBackward, SliceBackward示例