TensorFlow使用五个不同级别的日志消息。按照严重程度上升的顺序,它们是DEBUG
,INFO
,WARN
,ERROR
和FATAL
。当你配置任何级别日志记录时,TensorFlow将会输出这个级别(以及更高级别)的所有日志消息。例如,如果你设置的日志级别为ERROR
,你将获得包含ERROR
和FATAL
级别的日志信息, 如果你设置的日志级别为DEBUG
,你将得到所有五个级别的日志信息。
默认情况下,Tensorflow配置的日志信息级别为WARN
,但是当追踪模型训练时,你可能需要将日志信息级别调整为INFO
,这样的话,当你用fit
操作时,会获得额外的反馈。
将下面的代码添加到你的代码开始的地方(就在import S
之后):
tf.logging.set_verbosity(tf.logging.INFO)
如此这般,当你运行代码时,将看到如下所示的额外的日志输出:
INFO:tensorflow:loss = 1.18812, step = 1
INFO:tensorflow:loss = 0.210323, step = 101
INFO:tensorflow:loss = 0.109025, step = 201
通过INFO
级日志记录,tf.contrib.learn会在每100步后自动输出loss。
网友评论