| | |
|---|
| | | |
|---|
| | | 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 |
|---|
| | | ) |
|---|
| | | from funasr.models.frontend.abs_frontend import AbsFrontend |
|---|
| | | from funasr.models.specaug.abs_specaug import AbsSpecAug |
|---|
| | | from funasr.models.decoder.rnnt_decoder import RNNTDecoder |
|---|
| | |
|---|
| | | from funasr.models.encoder.abs_encoder import AbsEncoder |
|---|
| | | from funasr.models.joint_net.joint_network import JointNetwork |
|---|
| | | from funasr.modules.nets_utils import get_transducer_task_io |
|---|
| | | from funasr.modules.nets_utils import th_accuracy |
|---|
| | | from funasr.modules.add_sos_eos import add_sos_eos |
|---|
| | | from funasr.layers.abs_normalize import AbsNormalize |
|---|
| | | from funasr.torch_utils.device_funcs import force_gatherable |
|---|
| | | from funasr.models.base_model import FunASRModel |
|---|
| | |
|---|
| | | ) -> 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 |
|---|
| | |
|---|
| | | ) -> 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 |
|---|
