Commit 3c002d1c authored by Davis King's avatar Davis King

Made the steps without progress counter reset immediately upon changing the

learning rate.
parent 72a2e8e4
...@@ -381,7 +381,10 @@ namespace dlib ...@@ -381,7 +381,10 @@ namespace dlib
DLIB_CASSERT(lr > 0,""); DLIB_CASSERT(lr > 0,"");
wait_for_thread_to_pause(); wait_for_thread_to_pause();
if (learning_rate != lr) if (learning_rate != lr)
{
steps_without_progress = 0;
previous_loss_values.clear(); previous_loss_values.clear();
}
learning_rate = lr; learning_rate = lr;
lr_schedule.set_size(0); lr_schedule.set_size(0);
} }
......
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