NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made the steps without progress counter reset immediately upon changing the

Browse Source 
learning rate.
This commit is contained in:
Davis King 2016-06-05 07:45:15 -04:00
parent 72a2e8e437
commit 3c002d1cff
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dlib/dnn/trainer.h
View File

@ -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);
}