我有一种方法可以在 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