Skip to main content
 首页 » 编程设计

python之降低caffe培训输出的详细程度

2025年05月04日31kevingrace

我编译了带有调试标志的caffe。 现在当我跑的时候

./examples/mnist/train_lenet.sh 

我得到输出

I0112 22:50:49.680357 114020 data_layer.cpp:103]      Read time: 0.095 ms. 
I0112 22:50:49.680376 114020 data_layer.cpp:104] Transform time: 0.821 ms. 
I0112 22:50:49.681077 113921 solver.cpp:409]     Test net output #0: accuracy = 0.9902 
I0112 22:50:49.681115 113921 solver.cpp:409]     Test net output #1: loss = 0.0292544 (* 1 = 0.0292544 loss) 
I0112 22:50:49.681125 113921 solver.cpp:326] Optimization Done. 
I0112 22:50:49.681133 113921 caffe.cpp:215] Optimization Done. 
I0112 22:50:49.681915 114020 data_layer.cpp:102] Prefetch batch: 1 ms. 
I0112 22:50:49.681929 114020 data_layer.cpp:103]      Read time: 0.095 ms. 
I0112 22:50:49.681948 114020 data_layer.cpp:104] Transform time: 0.829 ms. 

http://pastebin.com/cbXTH5HH

我想要没有读取时间和预取时间的输出。无需重新编译。

I1130 00:30:48.030009  5007 solver.cpp:590] Iteration 9900, lr = 0.00596843 
I1130 00:30:49.105876  5007 solver.cpp:468] Snapshotting to binary proto file examples/mnist/lenet_iter_10000.caffemodel 
I1130 00:30:49.117113  5007 solver.cpp:753] Snapshotting solver state to binary proto file examples/mnist/lenet_iter_10000.solverstate 
I1130 00:30:49.125869  5007 solver.cpp:327] Iteration 10000, loss = 0.00332428 
I1130 00:30:49.125888  5007 solver.cpp:347] Iteration 10000, Testing net (#0) 
I1130 00:30:49.722595  5007 solver.cpp:415]     Test net output #0: accuracy = 0.9905 
I1130 00:30:49.722626  5007 solver.cpp:415]     Test net output #1: loss = 0.0302176 (* 1 = 0.0302176 loss) 
I1130 00:30:49.722642  5007 solver.cpp:332] Optimization Done. 
I1130 00:30:49.722647  5007 caffe.cpp:215] Optimization Done. 

例如:- http://pastebin.com/F5c3Yutu

请您参考如下方法:

怎么样

./examples/mnist/train_lenet.sh | grep -v "Read time:" | grep -v "Prefetch batch:" 

这给了我

I0112 22:50:49.671303 114020 data_layer.cpp:104] Transform time: 0.791 ms. 
I0112 22:50:49.672757 114020 data_layer.cpp:104] Transform time: 0.767 ms. 
I0112 22:50:49.674334 114020 data_layer.cpp:104] Transform time: 0.836 ms. 
I0112 22:50:49.675853 114020 data_layer.cpp:104] Transform time: 0.806 ms. 
I0112 22:50:49.677273 114020 data_layer.cpp:104] Transform time: 0.762 ms. 
I0112 22:50:49.678861 114020 data_layer.cpp:104] Transform time: 0.861 ms. 
I0112 22:50:49.680376 114020 data_layer.cpp:104] Transform time: 0.821 ms. 
I0112 22:50:49.681077 113921 solver.cpp:409]     Test net output #0: accuracy = 0.9902 
I0112 22:50:49.681115 113921 solver.cpp:409]     Test net output #1: loss = 0.0292544 (* 1 = 0.0292544 loss) 
I0112 22:50:49.681125 113921 solver.cpp:326] Optimization Done. 
I0112 22:50:49.681133 113921 caffe.cpp:215] Optimization Done. 
I0112 22:50:49.681948 114020 data_layer.cpp:104] Transform time: 0.829 ms. 

来自您的示例输入。