From 219c2482ab755fbd4e49dfbdee91bf1a8a4ec49a Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 19 五月 2023 11:33:27 +0800
Subject: [PATCH] websocket 2pass bugfix

---
 egs/aishell/transformer/utils/compute_cmvn.py |   10 +++++-----
 1 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/egs/aishell/transformer/utils/compute_cmvn.py b/egs/aishell/transformer/utils/compute_cmvn.py
index c57239a..949cc08 100755
--- a/egs/aishell/transformer/utils/compute_cmvn.py
+++ b/egs/aishell/transformer/utils/compute_cmvn.py
@@ -70,11 +70,11 @@
     parser = get_parser()
     args = parser.parse_args()
 
-    wav_scp_file = os.path.join(args.wav_path + "{}/wav.scp".format(args.idx))
-    cmvn_file = os.path.join(args.wav_path + "{}/cmvn.json".format(args.idx))
+    wav_scp_file = os.path.join(args.wav_path, "wav.{}.scp".format(args.idx))
+    cmvn_file = os.path.join(args.wav_path, "cmvn.{}.json".format(args.idx))
 
-    mean_stats = np.zeros(args.dims)
-    var_stats = np.zeros(args.dims)
+    mean_stats = np.zeros(args.dim)
+    var_stats = np.zeros(args.dim)
     total_frames = 0
 
     # with ReadHelper('ark:{}'.format(ark_file)) as ark_reader:
@@ -86,7 +86,7 @@
         lines = f.readlines()
         for line in lines:
             _, wav_file = line.strip().split()
-            fbank = compute_fbank(wav_file, num_mel_bins=args.dims)
+            fbank = compute_fbank(wav_file, num_mel_bins=args.dim)
             mean_stats += np.sum(fbank, axis=0)
             var_stats += np.sum(np.square(fbank), axis=0)
             total_frames += fbank.shape[0]

--
Gitblit v1.9.1