Support streaming conformer CTC models from wenet (#427)

2023-11-16 10:35:23 +08:00
parent b83b3e3cd1
commit fac4f6bc7c
31 changed files with 1212 additions and 7 deletions
--- a/scripts/wenet/export-onnx-streaming.py
+++ b/scripts/wenet/export-onnx-streaming.py
@@ -164,6 +164,7 @@ def main():
        dynamic_axes={
            "x": {0: "N", 1: "T"},
            "attn_cache": {2: "T"},
+            "attn_mask": {2: "T"},
            "log_probs": {0: "N"},
            "new_attn_cache": {2: "T"},
        },