我们常常会遇到一些问题,比如在tensorflow中如何屏蔽warning等问题,我们该怎么处理呢。下面这篇文章将为你提供一个解决思路,希望能帮你解决到相关问题。
1. 利用tf.logging.set_verbosity()函数
TensorFlow提供了一个tf.logging.set_verbosity()函数来设置logging的输出级别,可以通过设置输出级别来屏蔽warning。tf.logging.set_verbosity()函数接受一个参数,可以设置为tf.logging.ERROR、tf.logging.WARNING、tf.logging.INFO、tf.logging.DEBUG四个级别,分别表示只输出ERROR、输出ERROR和WARNING、输出ERROR、WARNING和INFO、输出ERROR、WARNING、INFO和DEBUG四种级别的logging输出。如果想要屏蔽warning,可以将参数设置为tf.logging.ERROR,即只输出ERROR级别的logging信息。
tf.logging.set_verbosity(tf.logging.ERROR)
2. 利用os.environ['TF_CPP_MIN_LOG_LEVEL']变量
此外,还可以利用os.environ['TF_CPP_MIN_LOG_LEVEL']变量来设置logging的输出级别,可以设置为'1'、'2'、'3'或者'0',分别表示只输出ERROR、输出ERROR和WARNING、输出ERROR、WARNING和INFO、输出ERROR、WARNING、INFO和DEBUG四种级别的logging输出。如果想要屏蔽warning,可以将参数设置为'1',即只输出ERROR级别的logging信息。
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'
3. 利用tf.compat.v1.logging.set_verbosity()函数
TensorFlow提供了一个tf.compat.v1.logging.set_verbosity()函数来设置logging的输出级别,可以通过设置输出级别来屏蔽warning。tf.compat.v1.logging.set_verbosity()函数接受一个参数,可以设置为tf.compat.v1.logging.ERROR、tf.compat.v1.logging.WARNING、tf.compat.v1.logging.INFO、tf.compat.v1.logging.DEBUG四个级别,分别表示只输出ERROR、输出ERROR和WARNING、输出ERROR、WARNING和INFO、输出ERROR、WARNING、INFO和DEBUG四种级别的logging输出。如果想要屏蔽warning,可以将参数设置为tf.compat.v1.logging.ERROR,即只输出ERROR级别的logging信息。
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
总结
以上就是为你整理的在tensorflow中如何屏蔽warning全部内容,希望文章能够帮你解决相关问题,更多请关注本站相关栏目的其它相关文章!