游雁
2024-06-06 f64bbaa036d5d82fc24c0c95d341849e4f1b4da6
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):