caffe 的边训练边测试

个人认为需要改以下两个地方:训练文件中增加测试层

                                                        在solver.prototxt中设定迭代多少次进行测试

train_val.prototxt

###### test layer, ignore_label

layer {
  name: "accuracy"
  type: "Accuracy"
  bottom: "score"
  bottom: "label"
  top: "accuracy"
  accuracy_param{ ignore_label: 255 }
  include {
    phase: TEST
  }
}



solver.prototxt
#################
net: "/home/Haichang_Li/study/caffe/models/lhc/train_val.prototxt"
##### we have test batch size 1, and 1449 test iterations.
test_iter: 1449
##### carry out testing every 50 training iterations. 
test_interval: 50
display: 10
average_loss: 10
lr_policy: "fixed"
# lr for unnormalized softmax -- see train_val definition
base_lr: 1e-4
# high momentum
momentum: 0.99
# no gradient accumulation
iter_size: 1
max_iter: 300000
weight_decay: 0.0005
snapshot: 10000
snapshot_prefix: "/home/Haichang_Li/study/caffe/models/lhc/train"
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值