如何在 Tensorflow (Python) 中抑制特定警告

Question

我有一个模型，基于某些条件，它有一些不相关的梯度，这正是我想要的。 但是 Tensorflow 每次遇到未连接的梯度时都会打印出警告。

WARNING:tensorflow:Gradients do not exist for variables

有没有办法只抑制这个特定的警告？ 我不想盲目地抑制所有警告，因为将来可能会出现意外（并且可能有用）的警告，因为我仍在处理我的模型。

Answer 1

有点骇人听闻的方式：

gradients = tape.gradient(loss, model.trainable_variables)
optimizer.apply_gradients([
    (grad, var) 
    for (grad, var) in zip(gradients, model.trainable_variables) 
    if grad is not None
])