码迷,mamicode.com
首页 > Web开发 > 详细

Caffe计算net、layer向前向后传播时间

时间:2017-04-26 11:50:15      阅读:371      评论:0      收藏:0      [点我收藏+]

标签:时间   mod   tools   log   命令   tool   caff   class   exe   

在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向前向后传播时间

标签:时间   mod   tools   log   命令   tool   caff   class   exe   

原文地址:http://www.cnblogs.com/ymjyqsx/p/6767628.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!