| | |
|---|
| | | from funasr.models.decoder.transformer_decoder import TransformerDecoder |
|---|
| | | from funasr.models.decoder.contextual_decoder import ContextualParaformerDecoder |
|---|
| | | from funasr.models.e2e_asr import ESPnetASRModel |
|---|
| | | from funasr.models.e2e_asr_paraformer import Paraformer, ParaformerBert, BiCifParaformer, ContextualParaformer |
|---|
| | | from funasr.models.e2e_asr_paraformer import Paraformer, ParaformerOnline, ParaformerBert, BiCifParaformer, ContextualParaformer |
|---|
| | | from funasr.models.e2e_tp import TimestampPredictor |
|---|
| | | from funasr.models.e2e_asr_mfcca import MFCCA |
|---|
| | | from funasr.models.e2e_uni_asr import UniASR |
|---|
| | |
|---|
| | | asr=ESPnetASRModel, |
|---|
| | | uniasr=UniASR, |
|---|
| | | paraformer=Paraformer, |
|---|
| | | paraformer_online=ParaformerOnline, |
|---|
| | | paraformer_bert=ParaformerBert, |
|---|
| | | bicif_paraformer=BiCifParaformer, |
|---|
| | | contextual_paraformer=ContextualParaformer, |
|---|
| | |
|---|
| | | type=str, |
|---|
| | | default="13_15", |
|---|
| | | help="The range of noise decibel level.", |
|---|
| | | ) |
|---|
| | | parser.add_argument( |
|---|
| | | "--batch_interval", |
|---|
| | | type=int, |
|---|
| | | default=10000, |
|---|
| | | help="The batch interval for saving model.", |
|---|
| | | ) |
|---|
| | | |
|---|
| | | for class_choices in cls.class_choices_list: |
|---|
