| | |
|---|
| | | |
|---|
| | | import torch |
|---|
| | | from packaging.version import parse as V |
|---|
| | | from typeguard import check_argument_types |
|---|
| | | from funasr.losses.label_smoothing_loss import ( |
|---|
| | | LabelSmoothingLoss, # noqa: H301 |
|---|
| | | ) |
|---|
| | |
|---|
| | | ) -> None: |
|---|
| | | """Construct an ESPnetASRTransducerModel object.""" |
|---|
| | | super().__init__() |
|---|
| | | |
|---|
| | | assert check_argument_types() |
|---|
| | | |
|---|
| | | # The following labels ID are reserved: 0 (blank) and vocab_size - 1 (sos/eos) |
|---|
| | | self.blank_id = 0 |
|---|
| | |
|---|
| | | """Construct an ESPnetASRTransducerModel object.""" |
|---|
| | | super().__init__() |
|---|
| | | |
|---|
| | | assert check_argument_types() |
|---|
| | | |
|---|
| | | # The following labels ID are reserved: 0 (blank) and vocab_size - 1 (sos/eos) |
|---|
| | | self.blank_id = 0 |
|---|
| | | |
|---|
| | |
|---|
| | | loss_lm = self._calc_lm_loss(decoder_out, target) |
|---|
| | | |
|---|
| | | loss_trans = loss_trans_utt + loss_trans_chunk |
|---|
| | | loss_ctc = loss_ctc + loss_ctc_chunk |
|---|
| | | loss_ctc = loss_ctc + loss_ctc_chunk |
|---|
| | | loss_ctc = loss_att + loss_att_chunk |
|---|
| | | |
|---|
| | | loss = ( |
|---|
| | |
|---|
| | | ignore_label=self.ignore_id, |
|---|
| | | ) |
|---|
| | | |
|---|
| | | return loss_att, acc_att |
|---|
| | | return loss_att, acc_att |
|---|
