mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-09 05:55:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update egs/reazonspeech/ASR/local/compute_fbank_musan.py

Browse Source 
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Bailey Machiko Hirota 2025-09-11 15:28:41 +09:00 committed by GitHub
parent a30e80cb0f
commit 9d389cdca7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/reazonspeech/ASR/local/compute_fbank_musan.py
View File

@ -94,12 +94,14 @@ def compute_fbank_musan(
logging.info("Extracting features for Musan")
if whisper_fbank:
device = "cuda" if torch.cuda.is_available() else "cpu"
if device == "cpu":
logging.warning("CUDA not available; using WhisperFbank on CPU.")
extractor = WhisperFbank(
WhisperFbankConfig(num_filters=num_mel_bins, device="cuda")
WhisperFbankConfig(num_filters=num_mel_bins, device=device)
)
else:
extractor = Fbank(FbankConfig(num_mel_bins=num_mel_bins))
with get_executor() as ex: # Initialize the executor only once.
# create chunks of Musan with duration 5 - 10 seconds
musan_cuts = (