This commit is contained in:
qipeng 2014-07-29 19:56:36 -07:00 коммит произвёл Jeff Donahue
Родитель 3f7a910e05
Коммит 23d4430869
2 изменённых файлов: 5 добавлений и 5 удалений

Просмотреть файл

@ -63,7 +63,7 @@ message NetParameter {
// NOTE
// Update the next available ID when you add a new SolverParameter field.
//
// SolverParameter next available ID: 27 (last added: test_state)
// SolverParameter next available ID: 31 (last added: delta)
message SolverParameter {
//////////////////////////////////////////////////////////////////////////////
// Specifying the train and test networks
@ -118,7 +118,7 @@ message SolverParameter {
optional float weight_decay = 12; // The weight decay.
// regularization types supported: L1 and L2
// controled by weight_decay
optional string regularization_type = 25 [default = "L2"];
optional string regularization_type = 28 [default = "L2"];
optional int32 stepsize = 13; // the stepsize for learning rate policy "step"
optional int32 snapshot = 14 [default = 0]; // The snapshot interval
optional string snapshot_prefix = 15; // The prefix for the snapshot.
@ -144,9 +144,9 @@ message SolverParameter {
NESTEROV = 1;
ADAGRAD = 2;
}
optional SolverType solver_type = 26 [default = SGD];
optional SolverType solver_type = 29 [default = SGD];
// numerical stability for AdaGrad
optional float delta = 27 [default = 1e-8];
optional float delta = 30 [default = 1e-8];
// If true, print information about the state of the net that may help with
// debugging learning problems.

Просмотреть файл

@ -510,7 +510,6 @@ void NesterovSolver<Dtype>::ComputeUpdateValue() {
vector<float>& net_params_weight_decay = this->net_->params_weight_decay();
// get the learning rate
Dtype rate = this->GetLearningRate();
Dtype delta = this->param_.delta();
if (this->param_.display() && this->iter_ % this->param_.display() == 0) {
LOG(INFO) << "Iteration " << this->iter_ << ", lr = " << rate;
}
@ -618,6 +617,7 @@ void AdaGradSolver<Dtype>::ComputeUpdateValue() {
vector<float>& net_params_weight_decay = this->net_->params_weight_decay();
// get the learning rate
Dtype rate = this->GetLearningRate();
Dtype delta = this->param_.delta();
if (this->param_.display() && this->iter_ % this->param_.display() == 0) {
LOG(INFO) << "Iteration " << this->iter_ << ", lr = " << rate;
}