mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-13 12:02:21 +00:00
Update infer.py
This commit is contained in:
parent
f9056bea91
commit
06c2993dfc
@ -183,13 +183,13 @@ def infer_dataset(
|
|||||||
sf.write(
|
sf.write(
|
||||||
file=params.save_wav_dir / f"{cut_ids[i]}_pred.wav",
|
file=params.save_wav_dir / f"{cut_ids[i]}_pred.wav",
|
||||||
data=output["waveform"],
|
data=output["waveform"],
|
||||||
samplerate=params.sampling_rate,
|
samplerate=params.data_args.sampling_rate,
|
||||||
subtype="PCM_16",
|
subtype="PCM_16",
|
||||||
)
|
)
|
||||||
sf.write(
|
sf.write(
|
||||||
file=params.save_wav_dir / f"{cut_ids[i]}_gt.wav",
|
file=params.save_wav_dir / f"{cut_ids[i]}_gt.wav",
|
||||||
data=audio[i].numpy(),
|
data=audio[i].numpy(),
|
||||||
samplerate=params.sampling_rate,
|
samplerate=params.data_args.sampling_rate,
|
||||||
subtype="PCM_16",
|
subtype="PCM_16",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user