Avoid NaN in feature normalization. (#2186)
This commit is contained in:
@@ -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);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user