首页 > 解决方案 > Tensorflow Hub 导致 tensorflow 日志记录重复!

问题描述

我的 TensorFlow 日志消息显示了两次。经过一番调查,我发现原因是 Tensorflow Hub。

例子:

代码:

import tensorflow as tf
import tensorflow_hub 

tf.logging.set_verbosity(tf.logging.INFO)

tf.logging.info("Hello test!")

输出:

INFO:tensorflow:Hello test!
I0523 16:35:51.024926 140735788589952 log.py:13] Hello test!

期望的输出:

 INFO:tensorflow:Hello test!

我尝试了什么:

我试图颠倒导入的顺序,最后只得到第二行输出。这更好,但我想知道如何只获得输出的第一行!谢谢你的帮助。

标签: machine-learningdeep-learningtensorflow

解决方案


我认为这里的问题是 Tensorflow Hub 使用https://abseil.io/docs/python/guides/logging是否作为 tensorflow 使用常规 python 日志记录。

切换 Tensorflow Hub 使用的日志记录类型是需要考虑的事情。同时,可以通过重新排序导入语句来解决此问题:

import tensorflow_hub as hub
import tensorflow as tf
tf.logging.set_verbosity(tf.logging.INFO)
tf.logging.info('This is a log')

推荐阅读