如何在Tensorflow（Python）中禁止显示特定警告

Question

我有一个模型，该模型根据某些条件具有一些未连接的渐变，而这正是我想要的。但是Tensorflow每次遇到未连接的渐变时都会打印出警告。

WARNING:tensorflow:Gradients do not exist for variables

是否有任何方法只能禁止显示此特定警告？我不想盲目地禁止所有警告，因为在我仍在研究模型时，将来可能会出现意外（且可能有用）的警告。

Answer 1

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