说明文档

简介

本仓库包含来自 NeMo 的 Citrinet-512 的 torchscript 模型。

参见 https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/stt_en_citrinet_512

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

citrinet = nemo_asr.models.EncDecCTCModelBPE.from_pretrained('stt_en_citrinet_512')
citrinet.export('model.onnx')

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

csukuangfj/sherpa-onnx-nemo-ctc-en-citrinet-512

作者 csukuangfj

↓ 0 ♥ 0

创建时间: 2023-04-06 12:12:36+00:00

更新时间: 2023-04-07 08:34:28+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