Caffe interface 之 Command Line

璇璇儿必须要努力

于 2015-10-15 14:53:24 发布

阅读量648

点赞数

分类专栏：学习笔记文章标签：深度学习-caffe

本文链接：https://blog.csdn.net/Forever__Xiaobai/article/details/49152559

版权

学习笔记专栏收录该内容

4 篇文章 0 订阅

订阅专栏

                    
                        
                    
                    在 
    
    CaffeRoot 
   下的/build/tools中运行caffe命令
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 2中的-gpu 2 中的2应该是GPU的编号，笔者只有一个GPU，将2改成0后，正常训练
caffe test -model examples/mnist/lenet_train_test.prototxt -weights examples/mnist/lenet_iter_10000.caffemodel -gpu 0 -iterations 100 
 首先，caffe是build/tools中的命令，test表明是要测试一个数据集。 
 -model 引出参数model的prototxt 
 -weight 引出训练得到的网络的参数输入 
 -GPU 引出使用的GPU编号 
 -iterations迭代次数 
  
 Note：迭代的次数是从开始的
caffe device_query -gpu 0：查看编号为0的GPU的状态

                