gpu bug fix

凌匀

2023-02-27 4c053ccc39ef3bdc6d131482274e9b1cd9ceee67

gpu bug fix

 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)