如何为张量流模型的特定层打印梯度值？

问题描述

我正在 TensorFlow 2.3 中训练模型

我想查看模型不同层的梯度值。

我知道如何访问模型的层。

with tf.GradientTape() as tape:
    lstm = model(song_emb_inp=song_emb_id_x_batch,time_bucket_emb_inp=time_bucket_emb_id_x_batch,training=True)
    
    loss = compute_loss(model,song_emb_id_y_batch,lstm)   
    
grads = tape.gradient(loss,model.trainable_variables)

for var,g in zip(model.trainable_variables,grads):
    print(f'{var.name},shape: {g.shape}')     

它给出了以下输出-

如何打印所有这些可训练层的梯度值？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）