OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Make sure the test loss the trainer logs to the console never gets suck at 

infinity.
This commit is contained in:
Davis King 2017-10-22 16:06:11 -04:00
parent 261f12d4ea
commit 04a8f0093d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dlib/dnn/trainer.h
View File

@ -584,7 +584,8 @@ namespace dlib
void record_test_loss(double loss) void record_test_loss(double loss)
{ {
test_previous_loss_values.push_back(loss); test_previous_loss_values.push_back(loss);
rs_test.add(loss); if (is_finite(loss))
rs_test.add(loss);
// discard really old loss values. // discard really old loss values.
while (test_previous_loss_values.size() > test_iter_without_progress_thresh) while (test_previous_loss_values.size() > test_iter_without_progress_thresh)
test_previous_loss_values.pop_front(); test_previous_loss_values.pop_front();