pytorch查看梯度

2024年5月8日 531次阅读来源: Nick.Zxx

打印一下梯度

            print('before backward ---------------------------------------')
            print(seq2seq.encoder.gru.weight_hh_l0.grad)
            loss.backward()
            print('after backward ---------------------------------------')
            print(seq2seq.encoder.gru.weight_hh_l0.grad)

纪念一下之前的一些疑惑

    原文作者：Nick.Zxx
    原文地址: https://zhuanlan.zhihu.com/p/36121066
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。