| | |
|---|
| | | name="specaug", |
|---|
| | | classes=dict( |
|---|
| | | specaug=SpecAug, |
|---|
| | | specaug_lfr=SpecAugLFR, |
|---|
| | | specaug_lfr=FSpecAugLR, |
|---|
| | | ), |
|---|
| | | type_check=AbsSpecAug, |
|---|
| | | default=None, |
|---|
| | |
|---|
| | | action=NestedDictAction, |
|---|
| | | default=get_default_kwargs(CTC), |
|---|
| | | help="The keyword arguments for CTC class.", |
|---|
| | | ) |
|---|
| | | group.add_argument( |
|---|
| | | "--joint_network_conf", |
|---|
| | | action=NestedDictAction, |
|---|
| | | default=None, |
|---|
| | | help="The keyword arguments for joint network class.", |
|---|
| | | ) |
|---|
| | | |
|---|
| | | group = parser.add_argument_group(description="Preprocess related") |
|---|
| | |
|---|
| | | num_optimizers: int = 1 |
|---|
| | | |
|---|
| | | class_choices_list = [ |
|---|
| | | model_choices, |
|---|
| | | frontend_choices, |
|---|
| | | specaug_choices, |
|---|
| | | normalize_choices, |
|---|
| | |
|---|
| | | try: |
|---|
| | | model_class = model_choices.get_class(args.model) |
|---|
| | | except AttributeError: |
|---|
| | | model_class = model_choices.get_class("asr") |
|---|
| | | model_class = model_choices.get_class("rnnt_unified") |
|---|
| | | |
|---|
| | | model = model_class( |
|---|
| | | vocab_size=vocab_size, |
|---|
