开发者社区> 问答> 正文

基于funasr构建离线长语音ASR,测试过程中发现内存泄漏,如何解决?

基于funasr构建了长语音识别服务,底层使用GPU进行推理。连续对100个时长为3小时的音频进行测试,发现内存一直在增长,所有任务结束后,内存也未释放。如何解决?
使用的模型如下:

        self.model = 'speech_paraformer-large-vad-punc-spk_asr_nat-zh-cn'
        self.vad_model = 'speech_fsmn_vad_zh-cn-16k-common-pytorch'
        self.punc_model = 'punc_ct-transformer_zh-cn-common-vocab272727-pytorch'
        self.spk_model = 'speech_campplus_sv_zh-cn_16k-common'

构建asr服务的方式如下:


asr_model = AutoModel(model=model_config.model,
                          vad_model=model_config.vad_model,
                          punc_model=model_config.punc_model,
                          spk_model=model_config.spk_model,
                          device='cuda:0')

推理过程如下:


asr_model.generate(input=audio, batchsize=1000)[0]

funasr版本为1.1.10

展开
收起
游客dtvjz4sonlg3u 2024-10-11 11:21:19 34 0
0 条回答
写回答
取消 提交回答
问答排行榜
最热
最新

相关电子书

更多
用AI 高效测试移动应用 立即下载
移动互联网测试到质量的转变 立即下载
给ITer的技术实战进阶课-阿里CIO学院独家教材(四) 立即下载