嘉渊 2 лет назад
Родитель
Сommit
89ddc6430d
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      egs/wenetspeech/conformer/local/process_opus.py

+ 1 - 1
egs/wenetspeech/conformer/local/process_opus.py

@@ -66,7 +66,7 @@ def output(output_wav_scp, utt_list, seg_path_list, start_time_list, end_time_li
             start = int(start_time_list[i] * sample_rate)
             end = int(end_time_list[i] * sample_rate)
             target_audio = waveform[:, start:end].transpose(0, 1).contiguous()
-            target_audio.export(seg_wav_path, format="wav")
+            torchaudio.save(seg_wav_path, target_audio, sample_rate)
 
             fout.write("{} {}\n".format(utt_id, seg_wav_path))
             if i % step == 0: