zhifu gao
2023-11-08 c3e2ca4b73d1d798cd7ec25263e06c357d9d9a2f
funasr/build_utils/build_pretrain_model.py
@@ -4,13 +4,18 @@
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(