Source code for espnet2.s2st.losses.ctc_loss

from typeguard import typechecked

from espnet2.s2st.losses.abs_loss import AbsS2STLoss


class S2STCTCLoss(AbsS2STLoss):
    """CTC-based loss for S2ST."""

    @typechecked
    def __init__(
        self,
        weight: float = 1.0,
    ):
        # Note(Jiatong): dummy CTC loss, only providing weight
        # for final loss calculation
        super().__init__()
        self.weight = weight

    def forward(loss):
        # dummy forward
        return loss