目录

一、准备工作

二、训练模型

三、训练期间的显示信息

一、准备工作

1.准备数据集之前请先切换到root模式，具体指令为：

sudo su
 

然后输入自己设定的密码

2.准备数据集，具体指令为：

  

   

    

     

    
    

     

      cd caffe/
     
    
   

    

     

    
    

     

      ./data/mnist/get_mnist.sh
     
    
   

    

     

    
    

     

      ./examples/mnist/create_mnist.sh
     
    
  
 

二、训练模型

1.因为下载的训练参数默认是GPU，我们需要将其改为CPU，否则报错，输入以下指令：

  

   

    

     

    
    

     

      cd caffe/examples/mnist/
     
    
   

    

     

    
    

     

      chmod 777 lenet_solver.prototxt
     
    
  
 

然后打开lenet_solver.prototxt，把最后一行GPU改为CPU

2.输入以下指令：

  

   

    

     

    
    

     

      cd caffe/
     
    
   

    

     

    
    

     

      ./examples/mnist/train_lenet.sh
     
    
  
 

这样就是开始训练模型了

三、训练期间的显示信息

1.参数和模式

2.这里最大迭代次数是10000，max_iter:10000，所以会到10000迭代

到最后结束 显示optimization done

3.最后生成的model叫lenet_iter_10000，之后就可以开始测试自己的数据集了！

文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。

原文链接：nickhuang1996.blog.csdn.net/article/details/79360552