From f13068082077e2a652d645a5d7630157ec881019 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 25 四月 2023 19:20:13 +0800
Subject: [PATCH] update

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

diff --git a/egs/aishell/transformer/utils/compute_cmvn.py b/egs/aishell/transformer/utils/compute_cmvn.py
index c57239a..4fefd3a 100755
--- a/egs/aishell/transformer/utils/compute_cmvn.py
+++ b/egs/aishell/transformer/utils/compute_cmvn.py
@@ -73,8 +73,8 @@
     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