mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-12 11:32:19 +00:00
Update infer.py
This commit is contained in:
parent
6e5c3e4032
commit
19f386ccf7
@ -179,16 +179,16 @@ def infer_dataset(
|
||||
output["waveform"] = to_waveform(output["mel"], vocoder, denoiser)
|
||||
|
||||
sf.write(
|
||||
file=params.save_wave_dir / f"{cut_ids[i]}_pred.wav",
|
||||
data=output["waveform"],
|
||||
samplerate=params.sampling_rate,
|
||||
subtype="PCM_16"
|
||||
file=params.save_wave_dir / f"{cut_ids[i]}_pred.wav",
|
||||
data=output["waveform"],
|
||||
samplerate=params.sampling_rate,
|
||||
subtype="PCM_16",
|
||||
)
|
||||
sf.write(
|
||||
file=params.save_wave_dir / f"{cut_ids[i]}_gt.wav",
|
||||
data=audio[i].numpy(),
|
||||
samplerate=params.sampling_rate,
|
||||
subtype="PCM_16"
|
||||
file=params.save_wave_dir / f"{cut_ids[i]}_gt.wav",
|
||||
data=audio[i].numpy(),
|
||||
samplerate=params.sampling_rate,
|
||||
subtype="PCM_16",
|
||||
)
|
||||
|
||||
num_cuts += batch_size
|
||||
|
Loading…
x
Reference in New Issue
Block a user