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

do a change

Browse Source
This commit is contained in:
luomingshuang 2022-04-08 19:12:30 +08:00
parent d8907769ee
commit 37f4f65ca3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py
View File

@ -98,7 +98,7 @@ def compute_fbank_wenetspeech_splits(args):
device = torch.device("cpu")
if torch.cuda.is_available():
device = torch.device("cuda", 5)
device = torch.device("cuda", 0)
extractor = KaldifeatFbank(KaldifeatFbankConfig(device=device))
logging.info(f"device: {device}")