소스 검색

Dev aky2 (#561)

* support resume model from pai

* add padding for streaming rnnt conv input

* fix large dataset training bug

* bug fix

* modify aishell rnnt egs to support wav input

* add libri_100 rnnt recipe

* bug fix

---------

Co-authored-by: aky15 <ankeyu.aky@11.17.44.249>
aky15 2 년 전
부모
커밋
fe63877bc8
1개의 변경된 파일19개의 추가작업 그리고 0개의 파일을 삭제
  1. 19 0
      funasr/utils/kwargs2args.py

+ 19 - 0
funasr/utils/kwargs2args.py

@@ -0,0 +1,19 @@
+import inspect
+
+
+def func(a: int, b, *, c, **kwargs):
+    pass
+
+def kwargs2args(func, kwargs):
+    parameters = inspect.signature(func).parameters
+    d = {k: i for i, k in enumerate(parameters)}
+    args = [None for i in range(len(parameters))]
+    for k, v in kwargs.items():
+        if k in d:
+            args[d[k]] = v
+
+    for i, v in enumerate(args):
+        if v is None:
+            break
+
+    return tuple(args[:i])