| | |
|---|
| | | import numpy as np |
|---|
| | | |
|---|
| | | import torch |
|---|
| | | from typeguard import check_argument_types |
|---|
| | | |
|---|
| | | from funasr.layers.abs_normalize import AbsNormalize |
|---|
| | | from funasr.models.ctc import CTC |
|---|
| | |
|---|
| | | frontend: Optional[AbsFrontend], |
|---|
| | | specaug: Optional[AbsSpecAug], |
|---|
| | | normalize: Optional[AbsNormalize], |
|---|
| | | preencoder: Optional[AbsPreEncoder], |
|---|
| | | encoder: AbsEncoder, |
|---|
| | | postencoder: Optional[AbsPostEncoder], |
|---|
| | | decoder: AbsDecoder, |
|---|
| | | ctc: CTC, |
|---|
| | | ctc_weight: float = 0.5, |
|---|
| | |
|---|
| | | crit_attn_weight: float = 0.0, |
|---|
| | | crit_attn_smooth: float = 0.0, |
|---|
| | | bias_encoder_dropout_rate: float = 0.0, |
|---|
| | | preencoder: Optional[AbsPreEncoder] = None, |
|---|
| | | postencoder: Optional[AbsPostEncoder] = None, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | assert 0.0 <= ctc_weight <= 1.0, ctc_weight |
|---|
| | | assert 0.0 <= interctc_weight < 1.0, interctc_weight |
|---|
| | | |
|---|
