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
DLIB_CASSERT(lr > 0,"");
wait_for_thread_to_pause();
if (learning_rate != lr)
{
steps_without_progress = 0;
previous_loss_values.clear();
}
learning_rate = lr;
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