time工具使用
命令:(以lenet为例)
${CAFF_ROOT}/build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt &> mnist_time.txt
注意,此时time工具的测试的时间,是训练的时间,包括forward和backward两部分。打开res.txt可以查看具体信息。
cpu版的BLCV/caffe,会调用多核版的blas库(mkl/openblas/atlas blas),可以通过设置线程数,分别测试不同线程情况下的性能。默认情况下,线程数为1.
设置线程数(环境变量的方式):
OMP_NUM_THREADS=24
export OMP_NUM_THREADS
echo $OMP_NUM_THREADS
若要测试alexnet,可用如下命令:
./build/tools/caffe.bin time -model ${CAFF_ROOT}/models/bvlc_alexnet/deploy.prototxt &> axlet_time.txt