From f9c99032c395f712e2bc067f8752a106f90f0650 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 8 May 2025 11:22:47 +0800 Subject: [PATCH] Avoid NaN in feature normalization. (#2186) --- sherpa-onnx/csrc/offline-stream.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/sherpa-onnx/csrc/offline-stream.cc b/sherpa-onnx/csrc/offline-stream.cc index 6fa0ca6c..c8263ea5 100644 --- a/sherpa-onnx/csrc/offline-stream.cc +++ b/sherpa-onnx/csrc/offline-stream.cc @@ -49,6 +49,11 @@ static void ComputeMeanAndInvStd(const float *p, int32_t num_rows, (*mean)[i] = t; float stddev = std::sqrt(sum_sq[i] / num_rows - t * t); + + if (stddev != stddev) { + stddev = 0; + } + (*inv_stddev)[i] = 1.0f / (stddev + 1e-5f); } }