From 5b90bc84c372298da56d2986da21b9ce8a1e5215 Mon Sep 17 00:00:00 2001 From: sushreebarsa <84765720+sushreebarsa@users.noreply.github.com> Date: Fri, 1 Dec 2023 13:02:28 +0530 Subject: [PATCH] Fixed typos in model doc --- official/nlp/modeling/ops/beam_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official/nlp/modeling/ops/beam_search.py b/official/nlp/modeling/ops/beam_search.py index 02d8634d146..3c3d31c49da 100644 --- a/official/nlp/modeling/ops/beam_search.py +++ b/official/nlp/modeling/ops/beam_search.py @@ -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])