标签: tensorflow keras
在使用fit或fit_generator方法时,我需要在训练期间实时监控梯度。这应该已经通过使用自定义callback函数来实现。但是,我无法正确访问渐变。属性model.optimizer.update返回梯度的张量,但需要提供数据。我想得到的是训练过程中最后一批应用的渐变值。
fit
fit_generator
callback
model.optimizer.update
以下答案没有给出相应的解决方案,因为它只是定义了一个通过提供额外数据来计算梯度的函数。
Getting gradient of model output w.r.t weights using Keras