Skip to content

Commit 50d246c

Browse files
committed
✍️ update message
1 parent 8e02e88 commit 50d246c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_asr/losses/rnnt_losses.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@
2222
try:
2323
from warprnnt_tensorflow import rnnt_loss as warp_rnnt_loss
2424
use_warprnnt = True
25+
print("Use RNNT loss in WarpRnnt")
2526
except ImportError:
26-
print("Cannot import RNNT loss in warprnnt. Falls back to RNNT in TensorFlow")
27-
print("Note: The RNNT in Tensorflow is not supported for CPU yet")
27+
print("Use RNNT loss in TensorFlow")
2828
use_warprnnt = False
2929

3030

0 commit comments

Comments
 (0)