Skip to content

Commit

Permalink
fixes after rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
qipeng authored and jeffdonahue committed Sep 1, 2014
1 parent 3f7a910 commit 23d4430
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions src/caffe/proto/caffe.proto
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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.
Expand All @@ -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.
Expand Down
2 changes: 1 addition & 1 deletion src/caffe/solver.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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;
}
Expand Down Expand Up @@ -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;
}
Expand Down

0 comments on commit 23d4430

Please sign in to comment.