mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-27 18:54:18 +00:00
add kaldifeatwhisper fbank
This commit is contained in:
parent
08db3051ad
commit
af29455c3d
@ -26,6 +26,8 @@ from lhotse import (
|
|||||||
CutSet,
|
CutSet,
|
||||||
WhisperFbank,
|
WhisperFbank,
|
||||||
WhisperFbankConfig,
|
WhisperFbankConfig,
|
||||||
|
KaldifeatWhisperFbank,
|
||||||
|
KaldifeatWhisperFbankConfig,
|
||||||
KaldifeatFbank,
|
KaldifeatFbank,
|
||||||
KaldifeatFbankConfig,
|
KaldifeatFbankConfig,
|
||||||
LilcomChunkyWriter,
|
LilcomChunkyWriter,
|
||||||
@ -128,9 +130,10 @@ def compute_fbank_wenetspeech_splits(args):
|
|||||||
if torch.cuda.is_available():
|
if torch.cuda.is_available():
|
||||||
device = torch.device("cuda", 0)
|
device = torch.device("cuda", 0)
|
||||||
if args.whisper_fbank:
|
if args.whisper_fbank:
|
||||||
extractor = WhisperFbank(
|
# extractor = WhisperFbank(
|
||||||
WhisperFbankConfig(num_filters=args.num_mel_bins, device=device)
|
# WhisperFbankConfig(num_filters=args.num_mel_bins, device=device)
|
||||||
)
|
# )
|
||||||
|
extractor = KaldifeatWhisperFbank(KaldifeatWhisperFbankConfig(num_filters=args.num_mel_bins, device=device))
|
||||||
else:
|
else:
|
||||||
extractor = KaldifeatFbank(KaldifeatFbankConfig(device=device))
|
extractor = KaldifeatFbank(KaldifeatFbankConfig(device=device))
|
||||||
logging.info(f"device: {device}")
|
logging.info(f"device: {device}")
|
||||||
|
Loading…
x
Reference in New Issue
Block a user