Skip to content

Commit f16fee4

Browse files
committed
✍️ fix ctc keras test step
1 parent 9144c7d commit f16fee4

File tree

1 file changed

+1
-1
lines changed
  • tensorflow_asr/models/keras

1 file changed

+1
-1
lines changed

tensorflow_asr/models/keras/ctc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def train_step(self, batch):
5656

5757
def test_step(self, batch):
5858
x, y_true = batch
59-
logit = self(x, training=False)
59+
logit = self(x["input"], training=False)
6060
y_pred = {
6161
"logit": logit,
6262
"logit_length": get_reduced_length(x["input_length"], self.time_reduction_factor)

0 commit comments

Comments
 (0)