speech_asr
2023-04-20 eac9f111b502e4581b14dc718731bf7dc1c7d5f6
funasr/utils/build_asr_model.py
@@ -210,7 +210,6 @@
    # frontend
    if args.input_size is None:
        # Extract features in the model
        frontend_class = frontend_choices.get_class(args.frontend)
        if args.frontend == 'wav_frontend':
            frontend = frontend_class(cmvn_file=args.cmvn_file, **args.frontend_conf)
@@ -218,7 +217,6 @@
            frontend = frontend_class(**args.frontend_conf)
        input_size = frontend.output_size()
    else:
        # Give features from data-loader
        args.frontend = None
        args.frontend_conf = {}
        frontend = None
@@ -347,5 +345,8 @@
    else:
        raise NotImplementedError("Not supported model: {}".format(args.model))
    # initialize
    if args.init is not None:
        initialize(model, args.init)
    return model