2016년 2월 16일 화요일

pycaffe로 logging하기

https://github.com/BVLC/caffe/pull/89 참고

1. 러닝시 발생하는 log를 파일로 가도록 redirection.
GLOG_logtostderr=1 ./build/examples/train_net.bin solver.prototxt 2> caffe.log
혹은 python 창에 뜨는 log를 copy & paste하여 파일에 저장함.(caffe.log)
단 이때, 저장되어야 할 log는 'I0217 ~~~'로 시작하는 caffe에서 발생한 log여야함. 다른 print로 발생한 log는 삭제해야됨.

2. 파일을 parsing하여 tr data의 loss와 vl data의 acc로 별도 파일로 만들어 냄.
parse_log.sh caffe.log [init_iter=0] [test_interval=1000]
위의 것을 써도 되고 혹은 python버전을 사용
https://github.com/BVLC/caffe/blob/master/tools/extra/parse_log.py

아님 caffe/tools/extra에 있음.

3. 각 파일을 draw
plot_training_log.py.example을 사용하면 됨.

결론:
1.번을 수행 후, caffe/tools/extra/plot_training_log.py.example을 사용하면 됨.