首页 > 代码库 > Caffe计算net、layer向前向后传播时间
Caffe计算net、layer向前向后传播时间
在caffe中计算某个model的整个net以及各个layer的向前向后传播时间，可以使用下面的命令格式：
./build/tools/caffe time --model=examples/mnist/mytest/execise1_train_test.prototxt -iterations 100 -gpu 0
其中model指向待测model的prototxt文件，-iterations后面的100表示迭代100次，-gpu 0 表示使用第一张显卡来计算时间，若使用CPU计算，将solver.prototxt的GPU改成CPU，并将-gpu 0去掉就行
./build/tools/caffe time --model=examples/mnist/mytest/execise1_train_test.prototxt -iterations 100
Caffe计算net、layer向前向后传播时间
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。