如何計算花費在CNN每個層上的花費時間和caffe界面？

Question

我試圖計算在fast-rcnn網絡中每一層花費的時間。我發現咖啡CMD界面caffe time可以做到這一點。但fast-rcnn是基於python代碼的，python界面中是否有任何函數可以幫助我計算時間？

Answer 1

python具有剖析工具。你可以看看timeit和cProfile工具。
如果你正在尋找一些更基本的，你可以使用time模塊測量時間：

import time 
t = time.time() 
run_rcnn_script() 
dt = time.time()-t 
print "time spent in run_rcnn_script: ", dt, " sec." 

Answer 2

要計算每一層

import timeit 

t1=timeit.default_timer() 
net.forward(start='start_layer_name',end='end_layer_name') 
t2=timeit.default_timer() 

print 'time is {}'.format(t2-t1) 

花費的時間