1. 程式人生 > >caffe測試訓練好的caffemodel和記錄每層的執行時間

caffe測試訓練好的caffemodel和記錄每層的執行時間

我用的是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”之類的來控制迴圈次數,如果是一個大型的模型的話。