Просмотр исходного кода

使用np.float32替换np.float (#789)

Yu Xin 2 лет назад
Родитель
Сommit
248a9979c3
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      funasr/models/frontend/wav_frontend.py

+ 2 - 2
funasr/models/frontend/wav_frontend.py

@@ -30,8 +30,8 @@ def load_cmvn(cmvn_file):
                 rescale_line = line_item[3:(len(line_item) - 1)]
                 vars_list = list(rescale_line)
                 continue
-    means = np.array(means_list).astype(np.float)
-    vars = np.array(vars_list).astype(np.float)
+    means = np.array(means_list).astype(np.float32)
+    vars = np.array(vars_list).astype(np.float32)
     cmvn = np.array([means, vars])
     cmvn = torch.as_tensor(cmvn, dtype=torch.float32)
     return cmvn