Merge pull request #2206 from yijinsheng/triton_gpu

Yuekai Zhang

2024-11-12 ef1d7b3f12535d73bf9046085d14d192c54cf7da

 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'],