Skip to content

Commit

Permalink
Fixed typos in model doc
Browse files Browse the repository at this point in the history
  • Loading branch information
sushreebarsa authored Dec 1, 2023
1 parent e4afd05 commit 5b90bc8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion official/nlp/modeling/ops/beam_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -587,7 +587,7 @@ def _gather_beams(nested, beam_indices, batch_size, new_beam_size):
Nested structure containing tensors with shape
[batch_size, new_beam_size, ...]
"""
# Computes the i'th coodinate that contains the batch index for gather_nd.
# Computes the i'th coordinate that contains the batch index for gather_nd.
# Batch pos is a tensor like [[0,0,0,0,],[1,1,1,1],..].
batch_pos = tf.range(batch_size * new_beam_size) // new_beam_size
batch_pos = tf.reshape(batch_pos, [batch_size, new_beam_size])
Expand Down

0 comments on commit 5b90bc8

Please sign in to comment.