基于pytorch深度学习项目所有参数梯度不更新情况

最近几天都被模型梯度不更新的情况困住,找到了一些关于模型参数输出的好方法
1、模型参数可视化
2、读取pth.tar文件好用的工具 netron 在线使用网址这个工具也可以下载到本地
3、好用的输出语句
调用的包名称
下面是显示的包的内容
在这里插入图片描述
还有就是在optimiter部分可以输出模型梯度
有关于模型梯度的输出参数有:
is_leaf
grad_fn
grad
模型参数不更新情况主要参照autobackward官方文档说明:

好用的模型参数输出语句

[x.grad for x in optimizer.param_groups[0]['params']]

最后我的梯度不更新的原因是因为环境问题:可能是因为在改bug的时候动了系统文件。

猜你喜欢

转载自blog.csdn.net/m0_45290027/article/details/127254827