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”之类的来控制循环次数,如果是一个大型的模型的话。
本文介绍如何使用Caffe工具进行模型测试及记录各层运行时间的方法。通过指定训练好的模型文件和权重文件,利用caffetest命令评估模型准确性;使用caffetime命令则能详细记录模型每一层的运行时间。
8341

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



