Caffe interface 之 Command Line

最新推荐文章于 2022-07-06 17:01:00 发布

璇璇儿必须要努力

最新推荐文章于 2022-07-06 17:01:00 发布

阅读量721

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/Forever__Xiaobai/article/details/49152559

学习笔记专栏收录该内容

4 篇文章

订阅专栏

本文介绍了如何使用Caffe命令进行模型训练和测试。包括在指定GPU上运行caffetrain和caffetest命令的方法，以及如何设置参数如模型文件、权重文件、GPU编号和迭代次数等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

                    
                    在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的状态