merged

782f4f48 · Davis King · 45ce9f7b · 04a8f009 · 782f4f48
Commit 782f4f48 authored Oct 23, 2017 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

dlib/dnn/trainer.h dlib/dnn/trainer.h +2 -1

No files found.
--- a/dlib/dnn/trainer.h
+++ b/dlib/dnn/trainer.h
@@ -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();