TensorFlow Logging 仅抑制警告消息

时间:2021-04-06 13:40:17

标签: python python-3.x tensorflow tensorflow2.0

我有一种方法可以在 Tensorflow 2 中仅抑制 warning 消息,同时保留 info 消息

我知道我们可以使用

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  

但在这种情况下,它将抑制此表后面的警告和信息消息。

<头>
级别 人类等级 级别说明
0 调试 [默认] 打印所有消息
1 信息 过滤掉 INFO 消息
2 警告 过滤掉信息和警告消息
3 错误 过滤掉所有消息

我需要保留显示有关我的训练状态的信息消息并禁止显示这些警告消息

INFO:tensorflow:Step 4300 per-step time 0.112s loss=3.361
WARNING:tensorflow:Ignoring ground truth with image id 1589478028 since it was previously added

0 个答案:

没有答案