caffe实例之mnist

来源:互联网 发布:黑暗之魂2猎龙枪数据 编辑:程序博客网 时间:2024/05/21 16:57

初次使用caffe框架,遇到了很多问题,虽说是小问题,但是细节不注意也是成功不了的。

1.路径一致:多个地方同一个文件的路径要一致

2.查bug的方法:因为caffa是基于C++的,支持Python,MATLAB,命令,写的是Python代码调用caffe,有些错误不会报出哎直接显示kenerl崩了,如何看到底是哪里报错了呢。

方法一:简历tmux,在终端用tmux attach命令可以查看最近的一个tmux,如果要查看的不是最近的一个tmux,则用ctrl+D关掉当前这个tmux 后再用tmux attach命令再次进入

方法二:登陆上服务器后,在命令窗口用jupyter notebook查看(jupter是一个在服务器上写代码的服务工具)

caffe是以深度学习的框架,

基于caffe的mnist主要分为以下几个步骤:

1.下载数据

2.将数据转化为lmdb格式,可通过调用caffe封装好的一个shell脚本完成

3.创建网络结构.prototxt格式,可手写也可通过代码编写生成,主要是网络由几层,每一层的类型以及size,map数量等等

4.创建solver文件,也是.prototxt格式,主要定义网络训练中的一些参数,比如学习率,batchsize,动量等参数。

5.网络初始化

6.训练

0 0
原创粉丝点击