如何使用Keras摆脱tensorflow详细消息

Question

我正在尝试张量流Keras后端.它不断向终端打印这些过于冗长的消息,这会破坏探测记录器的输出.如下.

h 1/200
   4608/3629568 [..............................] - ETA: 849s - loss: 1.1816I tensorflow/core/common_runtime/gpu/pool_allocator.cc:244] PoolAllocator: After 4208 get requests, put_count=4193 evicted_count=1000 eviction_rate=0.238493 and unsatisfied allocation rate=0.264971
I tensorflow/core/common_runtime/gpu/pool_allocator.cc:256] Raising pool_size_limit_ from 100 to 110

我该如何制作张量流量呢？我一直在浏览文档,我找不到类似.theanorc文件的设置.

Answer 1

如果您使用的是 TensorFlow 0.12，您可以设置TF_CPP_MIN_LOG_LEVEL环境变量来过滤日志消息。例如，您可以通过python这种方式开始以避免出现INFO-level 消息（例如任务中的“Raising pool_size_limit_”消息）：

$ TF_CPP_MIN_LOG_LEVEL=1 python ...