caffe测试训练好的caffemodel和记录每层的运行时间

来源:互联网 发布:淘宝违反价格法 编辑:程序博客网 时间:2024/06/05 17:13

我用的是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”之类的来控制循环次数,如果是一个大型的模型的话。

0 0