Unverified Commit bf410006 authored by Adrià Arrufat's avatar Adrià Arrufat Committed by GitHub
Browse files

Fix dnn_trainer trying to decrease the learning rate (#2442)

parent adca7472
...@@ -1061,7 +1061,7 @@ namespace dlib ...@@ -1061,7 +1061,7 @@ namespace dlib
// lower one instead. // lower one instead.
if (prob_loss_increasing_thresh >= prob_loss_increasing_thresh_max_value) if (prob_loss_increasing_thresh >= prob_loss_increasing_thresh_max_value)
{ {
if (verbose) if (verbose && learning_rate_shrink != 1)
std::cout << "(and while at it, also shrinking the learning rate)" << std::endl; std::cout << "(and while at it, also shrinking the learning rate)" << std::endl;
learning_rate = learning_rate_shrink * learning_rate; learning_rate = learning_rate_shrink * learning_rate;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment