Fix nemo feature normalization in test code (#2361)

This commit is contained in:
Fangjun Kuang
2025-07-08 15:41:56 +08:00
committed by GitHub
parent 831aff187d
commit f1405779cf
4 changed files with 9 additions and 14 deletions

View File

@@ -228,8 +228,8 @@ def main():
features = compute_features(audio, fbank)
if model.normalize_type != "":
assert model.normalize_type == "per_feature", model.normalize_type
mean = features.mean(axis=1, keepdims=True)
stddev = features.std(axis=1, keepdims=True) + 1e-5
mean = features.mean(axis=0, keepdims=True)
stddev = features.std(axis=0, keepdims=True) + 1e-5
features = (features - mean) / stddev
features = np.expand_dims(features, axis=0)