update

speech_asr

2023-04-20 eac9f111b502e4581b14dc718731bf7dc1c7d5f6

 funasr/utils/build_pretrain_model.py

@@ -57,7 +57,6 @@


def build_pretrain_model(args):
    if args.model_name == "data2vec":
        # frontend
        if args.input_size is None:
            frontend_class = frontend_choices.get_class(args.frontend)
@@ -90,6 +89,7 @@
            **args.encoder_conf,
        )

    if args.model_name == "data2vec":
        model_class = model_choices.get_class("data2vec")
        model = model_class(
            frontend=frontend,
@@ -97,8 +97,10 @@
            normalize=normalize,
            encoder=encoder,
        )
    else:
        raise NotImplementedError("Not supported model: {}".format(args.model))

        # 7. Initialize
    # initialize
        if args.init is not None:
            initialize(model, args.init)