Skip to content

Commit 794d3df

Browse files
Update examples/audio/md/transformer_asr.md
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 339b2d7 commit 794d3df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/audio/md/transformer_asr.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ class Transformer(keras.Model):
277277
preds = self([source, dec_input])
278278
one_hot = tf.one_hot(dec_target, depth=self.num_classes)
279279
mask = tf.math.logical_not(tf.math.equal(dec_target, 0))
280-
loss = self.compute_loss(None, one_hot, preds, sample_weight=mask)
280+
loss = self.compute_loss(y=one_hot, y_pred=preds, sample_weight=mask)
281281
self.loss_metric.update_state(loss)
282282
return {"loss": self.loss_metric.result()}
283283

0 commit comments

Comments
 (0)