Преглед изворни кода

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

Yu Xin пре 2 година
родитељ
комит
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)]
                 rescale_line = line_item[3:(len(line_item) - 1)]
                 vars_list = list(rescale_line)
                 vars_list = list(rescale_line)
                 continue
                 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 = np.array([means, vars])
     cmvn = torch.as_tensor(cmvn, dtype=torch.float32)
     cmvn = torch.as_tensor(cmvn, dtype=torch.float32)
     return cmvn
     return cmvn