"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "c0f88e04a075a9f1d811543149c11d1329245ae0"
Commit 3c002d1c authored by Davis King's avatar Davis King
Browse files

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