Caffe学习-caffe测试模型和统计运行时间

个人微信公众号:AI研习图书馆,欢迎关注~

深度学习知识及资源分享,学习交流,共同进步~

1. 测试模型

环境:Ubuntu16.04 LTS+caffe+cudn8
进入caffe目录:

cd caffe

(1)在terminal中运行命令

测试已经训练好的caffemodel的accuracy:

./build/tools/caffe test --model=/examples/mnist/lenet_train_test.prototxt --weights= /examples/mnist/lenet_iter_20000.caffemodel
#具体model和weights根据需要的model位置改变

(2)将命令放在.sh文件中运行
在这里插入图片描述
运行.sh脚本即可。

2. 统计运行时间

把上面的命令中test改成time可以记录每层用时:

./build/tools/caffe time --model=/examples/mnist/lenet_train_test.prototxt --weight = /examples/mnist/lenet_iter_20000.caffemodel

如果是一个大型的模型,可以在后面加上“–iteration=10”之类的来控制循环次数。

您的支持,是我不断创作的最大动力~

欢迎点赞关注留言交流~

深度学习,乐此不疲~

欢迎关注个人微信公众号~

Caffe学习-caffe测试模型和统计运行时间_第1张图片

你可能感兴趣的:(caffe学习)