Update infer.py

This commit is contained in:
zr_jin 2024-11-04 16:37:45 +08:00
parent 6e5c3e4032
commit 19f386ccf7

View File

@ -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