mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-13 12:02:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update infer.py

Browse Source
This commit is contained in:
zr_jin 2024-11-04 16:37:45 +08:00
parent 6e5c3e4032
commit 19f386ccf7
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/ljspeech/TTS/matcha/infer.py
View File

@ -182,13 +182,13 @@ def infer_dataset(
file=params.save_wave_dir / f"{cut_ids[i]}_pred.wav", file=params.save_wave_dir / f"{cut_ids[i]}_pred.wav",
data=output["waveform"], data=output["waveform"],
samplerate=params.sampling_rate, samplerate=params.sampling_rate,
subtype="PCM_16" subtype="PCM_16",
) )
sf.write( sf.write(
file=params.save_wave_dir / f"{cut_ids[i]}_gt.wav", file=params.save_wave_dir / f"{cut_ids[i]}_gt.wav",
data=audio[i].numpy(), data=audio[i].numpy(),
samplerate=params.sampling_rate, samplerate=params.sampling_rate,
subtype="PCM_16" subtype="PCM_16",
) )
num_cuts += batch_size num_cuts += batch_size