分类：打印梯度

2021
07-01

pytorch 如何打印网络回传梯度

需求：打印梯度，检查网络学习情况net=your_network().cuda()deftrain():...outputs=net(inputs)loss=criterion(outputs,targets)loss.backward()forname,parmsinnet.named_parameters():print('-->name:',name,'-->grad_requirs:',parms.requires_grad,\'-->grad_value:',parms.grad)...打印结果如下：name表示网络参数的名字；parms.requires_grad表示该参数是否可学习，是不是frozen的；p... 继续阅读 >