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加参:
temp: 0.5
}
如图:
在src/caffe/proto/caffe.proto 文件中,
optional float temp= 1 [default =0.5];
}
如图:
在include/caffe/vision_layers.hpp文件中添加变量:float softmax_temp;
如图:
如何使用?
请在SoftmaxWithLossLayer中,插入
softmax_temp=this->layer_param_.softmaxloss_param().temp();
如图: