NVIDIA-NeMo/Speech

Speed up RNNT model inference using TRT

开放

#14,531 创建于 2025年8月20日

 (1 条评论) (0 个反应) (1 位负责人)Python (3,561 个派生)github user discovery
ASRcommunity-requesthelp wantedwaiting-on-customer

仓库指标

星标
 (18,168 个星标)
PR 合并指标
 (30 天内没有已合并 PR)

描述

Hi,

I previously trained an RNNT model and now want to accelerate it by converting it to TensorRT. I’ve exported the model to ONNX and have encoder.onnx and decoder.onnx.

I’m using the TensorRT 25.03 Docker image and trtexec to convert the models. The decoder works fine with --fp16, but when I use --fp16 for the encoder, some outputs return NaN and the results are incorrect.

Has anyone encountered this issue or knows how to fix it?

Are there any methods to accelerate RNNT model inference?

贡献者指南