Fix error

23b841cf · pkufool · d55f5481 · 23b841cf
Commit 23b841cf authored Aug 11, 2022 by pkufool
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

fast_rnnt/python/fast_rnnt/mutual_information.py fast_rnnt/python/fast_rnnt/mutual_information.py +1 -1

No files found.
--- a/fast_rnnt/python/fast_rnnt/mutual_information.py
+++ b/fast_rnnt/python/fast_rnnt/mutual_information.py
@@ -287,7 +287,7 @@ def mutual_information_recursion(
            assert 0 <= t_begin <= t_end <= T

    # The following statements are for efficiency
-    px, py = px.is_contiguous(), py.is_contiguous()
+    px, py = px.contiguous(), py.contiguous()

    pxy_grads = [None, None]
    scores = MutualInformationRecursionFunction.apply(px, py, pxy_grads,