windowscaffe从0到1

首先要明确运行训练一个caffe模型需要哪几个文件

1、网络文件 train.prototxt文件,要修改训练图片、均值文件的路径,校验图片、均值文件的路径

2、solver.prototxt文件,要修改网络的路径和caffe模型的保存路径,我是存放在model4中,在运行程序之前要建好一个model4的文件夹;如果用的cpu把最后一行改成cpu

net: "/home/ths/MF_3_32_90/solver_cccp12_32/train_val_nofully21.prototxt"
test_iter: 60
test_interval: 1000
base_lr: 0.02
lr_policy: "inv"
gamma: 0.01
power: 0.75
#stepsize: 10000
display: 100
max_iter: 100000
momentum: 0.9
weight_decay: 0.0005
snapshot: 2000
snapshot_prefix: "/home/ths/MF_3_32_90/solver_cccp12_32/model4/train_val_cccp12_32input"
solver_mode: GPU

3、运行文件

如果用GPU:

train_caffenet.sh文件内容如下:要修改caffe的存放路径、solver.prototxt的路径、以及最后生成的log文件的存放路径

如果用CPU:

先编写一个run.txt的文件,写好后将后缀名改成bat,双击运行就可以了。具体示例在教程中

明确了训练一个caffe模型需要什么文件,知道每一步都是干什么之后就可以按照这个教程配caffe的环境了,这个教程写的很详细http://www.cnblogs.com/yixuan-xu/p/5858595.html


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值