This commit is contained in:
Davis King 2017-10-23 21:02:57 -04:00
commit 782f4f4825
1 changed files with 2 additions and 1 deletions

View File

@ -584,7 +584,8 @@ namespace dlib
void record_test_loss(double 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.
while (test_previous_loss_values.size() > test_iter_without_progress_thresh)
test_previous_loss_values.pop_front();