Fixed docs.

This commit is contained in:
yaozengwei 2022-04-21 19:55:30 +08:00
parent cf0ce8db32
commit d20a852f61

View File

@ -330,7 +330,7 @@ def greedy_search(
model: model:
The RNN-T model. The RNN-T model.
streams: streams:
A list of GreedySearchDecodingStream objects. A list of stream objects.
encoder_out: encoder_out:
A 3-D tensor of shape (N, T, encoder_out_dim) containing the output of A 3-D tensor of shape (N, T, encoder_out_dim) containing the output of
the encoder model. the encoder model.
@ -407,8 +407,8 @@ def modified_beam_search(
Args: Args:
model: model:
The RNN-T model. The RNN-T model.
stream: streams:
A stream object. A list of stream objects.
encoder_out: encoder_out:
A 3-D tensor of shape (N, T, encoder_out_dim) containing the output of A 3-D tensor of shape (N, T, encoder_out_dim) containing the output of
the encoder model. the encoder model.