您的位置：首页 > 理论基础 > 计算机网络

Caffe学习：使用pycaffe进行网络训练与测试

2015-09-04 21:11 781 查看

使用pycaffe进行网络训练与测试：

使用数据：密码：s9ol

编写solver文件：

train_net: "mnist/lenet_auto_train.prototxt"
test_net: "mnist/lenet_auto_test.prototxt"
test_iter: 100 # 每一次测试时进行100次forward pass
test_interval: 500
base_lr: 0.01
momentum: 0.9
weight_decay: 0.0005
lr_policy: "inv"
gamma: 0.0001
power: 0.75
display: 100
max_iter: 10000
snapshot: 5000
snapshot_prefix: "mnist/lenet"


设置Caffe使用GPU模式：

import caffe

caffe.set_device(0)
caffe.set_mode_gpu()


根据solver文件创建solver（使用SGDSolver）：

solver = caffe.SGDSolver('mnist/lenet_auto_solver.prototxt')


开始进行解算：

solver.solve()


即可实现训练与测试。
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航