ONNX 模型库
返回模型

说明文档

简介

本仓库包含来自 NeMo 的 stt_en_conformer_ctc_small 的 torchscript 模型。

参见 https://registry.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/stt_en_conformer_ctc_small

以下代码用于获取 model.onnx 和 tokens.txt:

m = nemo_asr.models.EncDecCTCModelBPE.from_pretrained('stt_en_conformer_ctc_small')
m.export('model.onnx')

with open('tokens.txt', 'w') as f:
  for i, s in enumerate(m.decoder.vocabulary):
    f.write(f\"{s} {i}\n\")
  f.write(f\"<blk> {i+1}\n\")

csukuangfj/sherpa-onnx-nemo-ctc-en-conformer-small

作者 csukuangfj

↓ 0 ♥ 1

创建时间: 2023-04-07 10:43:42+00:00

更新时间: 2023-04-07 11:00:58+00:00

在 Hugging Face 上查看

文件 (12)

.gitattributes
README.md
add-model-metadata.py
model.int8.onnx ONNX
model.onnx ONNX
quantize-model.py
test.py
test_wavs/0.wav
test_wavs/1.wav
test_wavs/8k.wav
test_wavs/trans.txt
tokens.txt