Merge pull request #4 from pkufool/fast_rnnt

Sync with k2 rnnt_loss

Merge pull request #4 from pkufool/fast_rnnt
Sync with k2 rnnt_loss
a283cddf · Daniel Povey · GitHub · b5828e2b · 182fe8de · a283cddf
Unverified Commit a283cddf authored Mar 16, 2022 by Daniel Povey Committed by GitHub Mar 16, 2022
12 changed files
--- a/fast_rnnt/python/tests/rnnt_loss_test.py
+++ b/fast_rnnt/python/tests/rnnt_loss_test.py
--- a/pyproject.toml
+++ b/pyproject.toml
-[build-system]
-requires = ['setuptools>=38.2.5', 'wheel', 'torch>=1.5', 'ninja']
-build-backend = "setuptools.build_meta"
\ No newline at end of file
--- a/setup.py
+++ b/setup.py
--- a/tests/requirements_test.txt
+++ b/tests/requirements_test.txt
-torch>=1.5
-tqdm
--- a/tests/test.py
+++ b/tests/test.py
--- a/torch_mutual_information/__init__.py
+++ b/torch_mutual_information/__init__.py
-from .mutual_information import mutual_information_recursion, joint_mutual_information_recursion
-from .rnnt import get_rnnt_logprobs, rnnt_loss_simple, rnnt_loss_aux
--- a/torch_mutual_information/mutual_information.py
+++ b/torch_mutual_information/mutual_information.py
--- a/torch_mutual_information/mutual_information_cpu.cpp
+++ b/torch_mutual_information/mutual_information_cpu.cpp
--- a/torch_mutual_information/mutual_information_cuda.cpp
+++ b/torch_mutual_information/mutual_information_cuda.cpp
--- a/torch_mutual_information/mutual_information_test.py
+++ b/torch_mutual_information/mutual_information_test.py
--- a/torch_mutual_information/rnnt.py
+++ b/torch_mutual_information/rnnt.py
--- a/torch_mutual_information/rnnt_test.py
+++ b/torch_mutual_information/rnnt_test.py