我们如何在Caffe库中找到Python层日志消息？

Question

Caffe库支持自定义Python层。为了调试该层，一种解决方案是在日志文件中输出一些内部变量，这是我所做的：

class MyLayer(caffe.Layer):
   def setup(self,bottom, top):
        import logging
        logging.info("hello the world")

在示例中，我想向日志文件中写入一些字符串“ hello the world”。

在定义了Python层之后，我们可以使用“ caffe train”程序来启动训练。但是，我在Caffe日志文件中找不到Python层日志信息（在Unbuntu中，它位于/tmp/caffe_username.log.INFO_date）。有任何想法吗？谢谢。

Answer 1

我相信您应该像其他Caffe一样使用Google日志记录glog库。参见此example：

import glog

#Simple L1 loss layer
class L1LossLayer(caffe.Layer):

    ...

    def forward(self, bottom, top):
        top[0].data[...] = ...
        glog.info('Loss is %f' % top[0].data[0])