fix: handling NaN propagation in BCE

Az-r-ow · May 17, 2024 · c9eff7c · c9eff7c
1 parent 1af5dea
commit c9eff7c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/NeuralNet/losses/BCE.hpp b/src/NeuralNet/losses/BCE.hpp
@@ -17,6 +17,10 @@ class BCE : public Loss {
         -(yTrim.array() * oTrim.array().log() +
           (1.0 - yTrim.array()) * (1.0 - oTrim.array()).log());
 
+    if (loss.array().isNaN().any())
+      throw std::runtime_error(
+          "NaN value encountered. Inputs might be too big");
+
     return loss.sum();
   }