Fix nemo feature normalization in test code (#2361)

2025-07-08 15:41:56 +08:00
parent 831aff187d
commit f1405779cf
4 changed files with 9 additions and 14 deletions
--- a/scripts/nemo/parakeet-tdt-0.6b-v2/test_onnx.py
+++ b/scripts/nemo/parakeet-tdt-0.6b-v2/test_onnx.py
@@ -233,8 +233,8 @@ def main():
    if model.normalize_type != "":
        assert model.normalize_type == "per_feature", model.normalize_type
        features = torch.from_numpy(features)
-        mean = features.mean(dim=1, keepdims=True)
-        stddev = features.std(dim=1, keepdims=True) + 1e-5
+        mean = features.mean(dim=0, keepdims=True)
+        stddev = features.std(dim=0, keepdims=True) + 1e-5
        features = (features - mean) / stddev
        features = features.numpy()
    print(audio.shape)