Skip to content

Commit

Permalink
correct naming in comment and message about average_loss
Browse files Browse the repository at this point in the history
  • Loading branch information
longjon committed Oct 6, 2014
1 parent cf27060 commit 914da95
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/caffe/proto/caffe.proto
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ message SolverParameter {
// the number of iterations between displaying info. If display = 0, no info
// will be displayed.
optional int32 display = 6;
// Display the cost averaged over the last average_cost iterations
// Display the loss averaged over the last average_loss iterations
optional int32 average_loss = 33 [default = 1];
optional int32 max_iter = 7; // the maximum number of iterations
optional string lr_policy = 8; // The learning rate decay policy.
Expand Down
2 changes: 1 addition & 1 deletion src/caffe/solver.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,7 @@ void Solver<Dtype>::Solve(const char* resume_file) {

int average_loss = this->param_.average_loss();

CHECK_GE(average_loss, 1) << "average_cost should be non-negative.";
CHECK_GE(average_loss, 1) << "average_loss should be non-negative.";

vector<Dtype> losses;
Dtype smoothed_loss = 0;
Expand Down

0 comments on commit 914da95

Please sign in to comment.