Shi Xian
2024-12-05 22b928dd3ff37ccee57ab2b5c2e4fcda4d33d24d
examples/industrial_data_pretraining/fsmn_kws_mt/convert.py
@@ -49,8 +49,6 @@
    copyfile(network_file, os.path.join(model_dir, 'origin.torch.pt'))
    model = FsmnKWSMTConvert(
        vocab_size=configs['encoder_conf']['output_dim'],
        vocab_size2=configs['encoder_conf']['output_dim2'],
        encoder='FSMNMTConvert',
        encoder_conf=configs['encoder_conf'],
        ctc_conf=configs['ctc_conf'],
@@ -82,8 +80,6 @@
    model_name="convert.torch.pt"
):
    model = FsmnKWSMTConvert(
        vocab_size=configs['encoder_conf']['output_dim'],
        vocab_size2=configs['encoder_conf']['output_dim2'],
        encoder='FSMNMTConvert',
        encoder_conf=configs['encoder_conf'],
        ctc_conf=configs['ctc_conf'],