zhifu gao
2024-05-08 a7bc099548576d6d855a3837523f51cadd0e8910
funasr/models/transformer/scorers/ctc.py
@@ -80,9 +80,7 @@
        """
        prev_score, state = state
        presub_score, new_st = self.impl(y.cpu(), ids.cpu(), state)
        tscore = torch.as_tensor(
            presub_score - prev_score, device=x.device, dtype=x.dtype
        )
        tscore = torch.as_tensor(presub_score - prev_score, device=x.device, dtype=x.dtype)
        return tscore, (presub_score, new_st)
    def batch_init_state(self, x: torch.Tensor):