zhifu gao
2023-02-27 2ed32a5d4014f80b5d0dac1b036b3fd391a10bd7
Merge pull request #162 from alibaba-damo-academy/dev_zly

gpu bug fix
1个文件已修改
1 ■■■■ 已修改文件
funasr/models/encoder/fsmn_encoder.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/encoder/fsmn_encoder.py
@@ -83,6 +83,7 @@
        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)