caffe修改

prototxt文件加参数实现:solver.prototxt加参和train_test.prototxt加参。

===========================================================================================================

转载自:http://blog.csdn.net/u014696921/article/details/52278581

solver.prototxt加参:

在solver.prototxt文件中添加变量:resultforsnap  = 0.64

如图:


在src/caffe/proto/caffe.proto 文件中,message SolverParameter {}中添加:optional float resultforsnap = 40 [default = 1e-8];


在include/caffe/solver.hpp文件中添加变量:float maxresult;

如图:



如何使用:在solver.cpp中调用?

示例为:我在初始化的函数中调用参数 maxresult = param_.resultforsnap();  

如图:


加完之后,先make clean一下,然后在make -j8,没错误标记调用成功。

===========================================================================================================

lb_train.prototxt加参:

在prototxt文件中,添加:
softmaxloss_param{
    temp: 0.5
 }
如图:

在src/caffe/proto/caffe.proto 文件中,

添加(如果已经有,就不用了):
optional SoftmaxLossParameter softmaxloss_param =41;
同时添加:
message SoftmaxLossParameter {
  optional float temp= 1 [default =0.5];
}
如图:

在include/caffe/vision_layers.hpp文件中添加变量:float softmax_temp;
如图:

如何使用?

请在SoftmaxWithLossLayer中,插入

softmax_temp=this->layer_param_.softmaxloss_param().temp();

如图:


然后make clean,最后make -j8,OK了!
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值