
如何使用PyTorch查看模型的权重(weight)和梯度(grad)信息?
本文共计969个文字,预计阅读时间需要4分钟。在用pdb调试时,有时需要查看特定layer的权重以及相应的梯度信息,如何查看呢?1. 首先打印出你的模型,例如:print(model)2.然后观察到model下面有module的key,mo
共收录篇相关文章

本文共计969个文字,预计阅读时间需要4分钟。在用pdb调试时,有时需要查看特定layer的权重以及相应的梯度信息,如何查看呢?1. 首先打印出你的模型,例如:print(model)2.然后观察到model下面有module的key,mo