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

Question

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

WARNING:tensorflow:Gradients do not exist for variables

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

Answer 1

有点hacky的方式：

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
])