speech_asr
2023-04-20 02f2a3c2ec07bfedb10cf6906c3df413fc8258f1
update
1个文件已修改
6 ■■■■■ 已修改文件
funasr/utils/build_model.py 6 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/utils/build_model.py
@@ -217,7 +217,7 @@
    encoder_class = encoder_choices.get_class(args.encoder)
    encoder = encoder_class(input_size=input_size, **args.encoder_conf)
    # 7. Decoder
    # decoder
    decoder_class = decoder_choices.get_class(args.decoder)
    decoder = decoder_class(
        vocab_size=vocab_size,
@@ -225,7 +225,9 @@
        **args.decoder_conf,
    )
    # 8. CTC
    # ctc
    ctc = CTC(
        odim=vocab_size, encoder_output_size=encoder.output_size(), **args.ctc_conf
    )