From 37f4f65ca3241a64fe861b16ba03a6dc8806a156 Mon Sep 17 00:00:00 2001
From: luomingshuang <739314837@qq.com>
Date: Fri, 8 Apr 2022 19:12:30 +0800
Subject: [PATCH] do a change

---
 egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py b/egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py
index 91058b066..9e290d7ca 100755
--- a/egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py
+++ b/egs/wenetspeech/ASR/local/compute_fbank_wenetspeech_splits.py
@@ -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}")