From 914da95ed1059c05d39d9b6fbe0f797970aaa376 Mon Sep 17 00:00:00 2001 From: Jonathan L Long Date: Mon, 6 Oct 2014 16:35:33 -0700 Subject: [PATCH] correct naming in comment and message about average_loss --- src/caffe/proto/caffe.proto | 2 +- src/caffe/solver.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto index 751b68232cd..f0404a09b90 100644 --- a/src/caffe/proto/caffe.proto +++ b/src/caffe/proto/caffe.proto @@ -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. diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp index 221fa07e464..995df06ddc7 100644 --- a/src/caffe/solver.cpp +++ b/src/caffe/solver.cpp @@ -171,7 +171,7 @@ void Solver::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 losses; Dtype smoothed_loss = 0;