在TensorFlow中屏蔽警告的方式有很多种,以下是两种常见的方式:
1. 禁止TensorFlow警告输出
在TensorFlow运行时会输出大量的警告信息,如果想要屏蔽这些警告信息,可以使用以下代码:
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
import tensorflow as tf
其中,os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
表示只输出TensorFlow的错误信息,将警告和其他信息屏蔽。需要在导入tensorflow
之前加入该代码。
2. 使用Python警告模块屏蔽TensorFlow警告
利用Python自带的警告模块可以屏蔽TensorFlow的警告,避免影响程序的运行。使用方式如下:
import warnings
warnings.filterwarnings('ignore')
import tensorflow as tf
其中,warnings.filterwarnings('ignore')
表示忽略所有警告。需要在导入tensorflow
之前加入该代码。
以上是在TensorFlow中屏蔽警告的两种常见方式,可以根据自己的需要选择其中一种使用。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:在TensorFlow中屏蔽warning的方式 - Python技术站