如何使用Keras计算损失对模型权重的梯度导数？

本文共计868个文字，预计阅读时间需要4分钟。

笔记：怀疑模型梯度爆炸，想打印模型 + loss + 各个权重的导数看看。如果fit来训练的话，可以使用keras.callbacks.TensorBoard实现。但此次使用train_on_batch来训练，需用K.gradients和K.function实现。

Notes

怀疑模型梯度爆炸，想打印模型 loss 对各权重的导数看看。如果如果fit来训练的话，可以用keras.callbacks.TensorBoard实现。

但此次使用train_on_batch来训练的，用K.gradients和K.function实现。