Luzhi6688

Results 3 comments of Luzhi6688

谢谢大佬的回复,我现在用的是SGD随机梯度下降,网络模型是LSTM,想提取每个客户机在本地训练完成以后的梯度信息,我发现弄了很久提取的只有模型的参数信息,如图, ![捕获](https://user-images.githubusercontent.com/100510083/181401566-9ae7a3c9-d3ce-4c2c-8315-9ba6678e8245.JPG),是我获取梯度的函数,不知道对不对,希望得到您的帮助,谢谢大佬! ![梯度信息](https://user-images.githubusercontent.com/100510083/181401776-2163ffc3-6321-4094-b88e-6860534583b7.JPG)

是用pytorch写的,把值输出来不知道到底是梯度值还是参数值,卡在这里很久了,一直解决不掉。

好的我再试一下,谢谢您的帮助!