caffe计算net、layer向前向后传播时间

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/csuzhaoqinghui/article/details/86645069

在caffe中计算某个model的整个net以及各个layer的向前向后传播时间,可以使用下面的命令格式:

./build/tools/caffe time --model=examples/mnist/mytest/execise1_train_test.prototxt -iterations 100 -gpu 0

其中model指向待测model的prototxt文件,-iterations后面的100表示迭代100次,-gpu 0 表示使用第一张显卡来计算时间,若使用CPU计算,将solver.prototxt的GPU改成CPU,并将-gpu 0去掉就行

./build/tools/caffe time --model=examples/mnist/mytest/execise1_train_test.prototxt -iterations 100

可以对比一下CPU与GPU的速度
--------------------- 
作者:qyxqyx123 
来源:CSDN 
原文:https://blog.csdn.net/qyxqyx123/article/details/53431724 
版权声明:本文为博主原创文章,转载请附上博文链接!

猜你喜欢

转载自blog.csdn.net/csuzhaoqinghui/article/details/86645069