虚拟机Ubuntu 16.04 64位 Caffe 训练LeNet 数据集MNIST

举报
悲恋花丶无心之人 发表于 2021/02/03 00:05:57 2021/02/03
3.1k+ 0 0
【摘要】 目录 一、准备工作 二、训练模型 三、训练期间的显示信息 一、准备工作 1.准备数据集之前请先切换到root模式,具体指令为: sudo su 然后输入自己设定的密码 2.准备数据集,具体指令为: cd caffe/./data/mnist/get_mnist.sh./examples/mnist/create_mnist.sh 二、训练模型 1.因...

目录

一、准备工作

二、训练模型

三、训练期间的显示信息


一、准备工作

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

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

作者其他文章

评论(0

抱歉,系统识别当前为高风险访问,暂不支持该操作

    全部回复

    上滑加载中

    设置昵称

    在此一键设置昵称,即可参与社区互动!

    *长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

    *长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。