-
Notifications
You must be signed in to change notification settings - Fork 2.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable stateful decoding of RNNT over multiple transcribe calls #3037
Conversation
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
Signed-off-by: smajumdar <[email protected]>
This pull request introduces 3 alerts when merging d7598fc into 1f36f32 - view on LGTM.com new alerts:
|
This pull request introduces 3 alerts when merging 6318e3f into 1f36f32 - view on LGTM.com new alerts:
|
This pull request introduces 3 alerts when merging 4b235eb into 9405273 - view on LGTM.com new alerts:
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
…IA#3037) * Start on stateful external decoding Signed-off-by: smajumdar <[email protected]> * Prepare connectors Signed-off-by: smajumdar <[email protected]> * Refactor greedy sample decoding to use Hypothesis Signed-off-by: smajumdar <[email protected]> * Refactor greedy batch first mode for Hypothesis Signed-off-by: smajumdar <[email protected]> * Update second case of greedy batch decoding Signed-off-by: smajumdar <[email protected]> * Start stateful decoding Signed-off-by: smajumdar <[email protected]> * Add guards for stateful decoding Signed-off-by: smajumdar <[email protected]> * Fix state management when no states is provided Signed-off-by: smajumdar <[email protected]> * Create Signed-off-by: smajumdar <[email protected]> * Correct logging Signed-off-by: smajumdar <[email protected]> * Begin support for stateful beam decoding Signed-off-by: smajumdar <[email protected]> * Update streaming utils with method 2 Signed-off-by: smajumdar <[email protected]> * Initiate stateful beam implementation Signed-off-by: smajumdar <[email protected]> * Reset changes Signed-off-by: smajumdar <[email protected]> * Fix style Signed-off-by: smajumdar <[email protected]> Co-authored-by: Jagadeesh Balam <[email protected]>
* Start on stateful external decoding Signed-off-by: smajumdar <[email protected]> * Prepare connectors Signed-off-by: smajumdar <[email protected]> * Refactor greedy sample decoding to use Hypothesis Signed-off-by: smajumdar <[email protected]> * Refactor greedy batch first mode for Hypothesis Signed-off-by: smajumdar <[email protected]> * Update second case of greedy batch decoding Signed-off-by: smajumdar <[email protected]> * Start stateful decoding Signed-off-by: smajumdar <[email protected]> * Add guards for stateful decoding Signed-off-by: smajumdar <[email protected]> * Fix state management when no states is provided Signed-off-by: smajumdar <[email protected]> * Create Signed-off-by: smajumdar <[email protected]> * Correct logging Signed-off-by: smajumdar <[email protected]> * Begin support for stateful beam decoding Signed-off-by: smajumdar <[email protected]> * Update streaming utils with method 2 Signed-off-by: smajumdar <[email protected]> * Initiate stateful beam implementation Signed-off-by: smajumdar <[email protected]> * Reset changes Signed-off-by: smajumdar <[email protected]> * Fix style Signed-off-by: smajumdar <[email protected]> Co-authored-by: Jagadeesh Balam <[email protected]> Signed-off-by: PeganovAnton <[email protected]>
…IA#3037) * Start on stateful external decoding Signed-off-by: smajumdar <[email protected]> * Prepare connectors Signed-off-by: smajumdar <[email protected]> * Refactor greedy sample decoding to use Hypothesis Signed-off-by: smajumdar <[email protected]> * Refactor greedy batch first mode for Hypothesis Signed-off-by: smajumdar <[email protected]> * Update second case of greedy batch decoding Signed-off-by: smajumdar <[email protected]> * Start stateful decoding Signed-off-by: smajumdar <[email protected]> * Add guards for stateful decoding Signed-off-by: smajumdar <[email protected]> * Fix state management when no states is provided Signed-off-by: smajumdar <[email protected]> * Create Signed-off-by: smajumdar <[email protected]> * Correct logging Signed-off-by: smajumdar <[email protected]> * Begin support for stateful beam decoding Signed-off-by: smajumdar <[email protected]> * Update streaming utils with method 2 Signed-off-by: smajumdar <[email protected]> * Initiate stateful beam implementation Signed-off-by: smajumdar <[email protected]> * Reset changes Signed-off-by: smajumdar <[email protected]> * Fix style Signed-off-by: smajumdar <[email protected]> Co-authored-by: Jagadeesh Balam <[email protected]>
Changelog
transcribe
now takespartial_hypotheses
as optional input, provided in returned hypothesis from previous transcribe callbatch_concat_states
andbatch_copy_states
support for RNNT models to abstract away state management.y_sequence
: tensor form of int ids of tokensdec_state
: CPU tuple of decoder statesscore
: logprob score of current y_sequencealignments
: 2d dangling matrix of token alignmentsExample