| | |
|---|
| | | from funasr.models.encoder.data2vec_encoder import Data2VecEncoder |
|---|
| | | from funasr.models.frontend.default import DefaultFrontend |
|---|
| | | from funasr.models.frontend.windowing import SlidingWindow |
|---|
| | | from funasr.models.frontend.wav_frontend import WavFrontend |
|---|
| | | from funasr.models.specaug.specaug import SpecAug |
|---|
| | | from funasr.torch_utils.initialize import initialize |
|---|
| | | from funasr.train.class_choices import ClassChoices |
|---|
| | | |
|---|
| | | frontend_choices = ClassChoices( |
|---|
| | | name="frontend", |
|---|
| | | classes=dict(default=DefaultFrontend, sliding_window=SlidingWindow), |
|---|
| | | classes=dict( |
|---|
| | | default=DefaultFrontend, |
|---|
| | | sliding_window=SlidingWindow, |
|---|
| | | wav_frontend=WavFrontend, |
|---|
| | | ), |
|---|
| | | default="default", |
|---|
| | | ) |
|---|
| | | specaug_choices = ClassChoices( |
|---|
| | |
|---|
| | | **args.encoder_conf, |
|---|
| | | ) |
|---|
| | | |
|---|
| | | if args.model_name == "data2vec": |
|---|
| | | if args.model == "data2vec": |
|---|
| | | model_class = model_choices.get_class("data2vec") |
|---|
| | | model = model_class( |
|---|
| | | frontend=frontend, |
|---|
