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

gpu bug fix

Browse Source
This commit is contained in:
凌匀 2023-02-27 19:10:15 +08:00
parent 4e4eed605e
commit 4c053ccc39
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/models/encoder/fsmn_encoder.py
View File

@ -82,7 +82,8 @@ class FSMNBlock(nn.Module):
def forward(self, input: torch.Tensor, cache: torch.Tensor):
x = torch.unsqueeze(input, 1)
x_per = x.permute(0, 3, 2, 1) # B D T C
cache = cache.to(x_per.device)
y_left = torch.cat((cache, x_per), dim=2)
cache = y_left[:, :, -(self.lorder - 1) * self.lstride:, :]
y_left = self.conv_left(y_left)
@ -297,4 +298,4 @@ if __name__ == '__main__':
print('input shape: {}'.format(x.shape))
print('output shape: {}'.format(y.shape))
print(fsmn.to_kaldi_net())
print(fsmn.to_kaldi_net())