Update infer.py

This commit is contained in:
zr_jin 2024-11-05 11:45:32 +08:00
parent f9056bea91
commit 06c2993dfc

View File

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