如何在 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
])