caffe安装成功后测试

mnist模型测试

参考博客

1.准备数据

cd caffe
sudo ./data/mnist/get_mnist.sh #下载数据集
sudo ./examples/mnist/create_mnist.sh #转换格式

在这里插入图片描述

2.训练

sudo ./examples/mnist/train_lenet.sh

出现错误:

wangwh@wangwh-Lenovo-YOGA-710-14IKB:~/caffe$ sudo ./examples/mnist/train_lenet.sh
I1220 10:55:37.676761 20412 caffe.cpp:204] Using GPUs 0
F1220 10:55:37.676882 20412 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode.
*** Check failure stack trace: ***
    @     0x7fe4c4c685cd  google::LogMessage::Fail()
    @     0x7fe4c4c6a433  google::LogMessage::SendToLog()
    @     0x7fe4c4c6815b  google::LogMessage::Flush()
    @     0x7fe4c4c6ae1e  google::LogMessageFatal::~LogMessageFatal()
    @     0x7fe4c51a8a60  caffe::Caffe::SetDevice()
    @           0x40a78b  train()
    @           0x407208  main
    @     0x7fe4c344a830  __libc_start_main
    @           0x407ad9  _start
    @              (nil)  (unknown)
Aborted (core dumped)

分析:因为我没有配置GPU,所以需要修改这个文件。

  • 尝试解决:
    修改了文件lenet_solver.prototxt:
# solver mode: CPU or GPU
solver_mode: CPU 

发现可以训练了,这不简单的嘛哈哈哈,不知道训练得多长时间啊。坏了,忘记修改一下迭代次数了,这只是练习而已啊,表示一下就行了。。。
成功了:
在这里插入图片描述竟然是经过了一万次迭代,有点多了,刚才改成1000次还快一些。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值