Как найти сообщение журнала слоя Python в библиотеке Caffe? - PullRequest
0 голосов
/ 25 февраля 2019

Библиотека Caffe поддерживает самоопределяемый слой Python.Чтобы отладить этот слой, одним из решений является вывод некоторой внутренней переменной в файл журнала, и вот что я сделал:

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

В этом примере я хочу написать несколько строк: "Привет, мир!""в файл журнала.

После того, как слой Python определен, мы можем использовать программу «caffe train» для запуска обучения.Однако я не могу найти информацию журнала уровня Python в файле журнала Caffe (в Unbuntu она находится по адресу /tmp/caffe_username.log.INFO_date).Есть идеи?Благодарю.

1 Ответ

0 голосов
/ 25 февраля 2019

Я полагаю, что вы должны использовать библиотеку Google logging glog, как и остальная часть Caffe.Посмотрите этот пример :

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])
...