speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
语帆 2024-02-28 15:31:14 +08:00
parent debafeac37
commit 52fee96d71
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/auto/auto_model.py
View File

@ -141,7 +141,7 @@ class AutoModel:
kwargs = download_model(**kwargs) kwargs = download_model(**kwargs)
set_all_random_seed(kwargs.get("seed", 0)) set_all_random_seed(kwargs.get("seed", 0))
pdb.set_trace()
device = kwargs.get("device", "cuda") device = kwargs.get("device", "cuda")
if not torch.cuda.is_available() or kwargs.get("ngpu", 1) == 0: if not torch.cuda.is_available() or kwargs.get("ngpu", 1) == 0:
device = "cpu" device = "cpu"

3
funasr/models/lcbnet/model.py
View File

@ -274,15 +274,12 @@ class LCBNet(nn.Module):
ind: int ind: int
""" """
with autocast(False): with autocast(False):
pdb.set_trace()
# Data augmentation # Data augmentation
if self.specaug is not None and self.training: if self.specaug is not None and self.training:
speech, speech_lengths = self.specaug(speech, speech_lengths) speech, speech_lengths = self.specaug(speech, speech_lengths)
pdb.set_trace()
# Normalization for feature: e.g. Global-CMVN, Utterance-CMVN # Normalization for feature: e.g. Global-CMVN, Utterance-CMVN
if self.normalize is not None: if self.normalize is not None:
speech, speech_lengths = self.normalize(speech, speech_lengths) speech, speech_lengths = self.normalize(speech, speech_lengths)
pdb.set_trace()
# Forward encoder # Forward encoder
# feats: (Batch, Length, Dim) # feats: (Batch, Length, Dim)
# -> encoder_out: (Batch, Length2, Dim2) # -> encoder_out: (Batch, Length2, Dim2)