我用的是ubuntu的caffe。
在caffe的目录下
在terminal中运行以下命令用以测试已经训练好的caffemodel的accuracy:
(或者将以下命令放在.sh文件中运行)
./build/tools/caffe test --model=/examples/mnist/lenet_train_test.prototxt --weights= /examples/mnist/lenet_iter_20000.caffemodel
#具体model和weights根据需要的model位置改变
类似地,如果用把上面的命令中test改成time可以记录每层用时
./build/tools/caffe time --model=/examples/mnist/lenet_train_test.prototxt --weight = /examples/mnist/lenet_iter_20000.caffemodel
可以在后面加上“--iteration=10”之类的来控制循环次数,如果是一个大型的模型的话。
本文介绍如何在Ubuntu环境下使用Caffe工具测试已训练模型的准确率与各层执行时间。通过具体命令示例展示如何调整迭代次数来优化大型模型的测试过程。

795

被折叠的 条评论
为什么被折叠?



