您的位置:首页 > Web前端

caffe在prototxt文件加参数

2016-01-08 20:43 585 查看
prototxt文件加参数实现:solver.prototxt加参和train_test.prototxt加参。

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

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了!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: