启用TensorFlow的日志记录(Enabling Logging with TensorFlow)

TensorFlow使用五个不同级别的日志消息。按照严重程度上升的顺序,它们是DEBUGINFOWARNERRORFATAL。当你配置任何级别日志记录时,TensorFlow将会输出这个级别(以及更高级别)的所有日志消息。例如,如果你设置的日志级别为ERROR,你将获得包含ERRORFATAL级别的日志信息, 如果你设置的日志级别为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。

你可能感兴趣的:(启用TensorFlow的日志记录(Enabling Logging with TensorFlow))