NVIDIA-NeMo/Speech

Speed up RNNT model inference using TRT

開放

#14,531 建立於 2025年8月20日

 (1 則留言) (0 個反應) (1 位負責人)Python (3,527 個分叉)github user discovery
ASRcommunity-requesthelp wantedwaiting-on-customerwaiting-on-maintainers

倉庫指標

星標
 (17,872 顆星)
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?

貢獻者指南