greedy_decode: Greedy Decoding
Description
Greedy Decoding
Usage
greedy_decode(
model,
encoder_output,
initial_tokens,
tokenizer,
max_length = 448L,
device
)
Value
Integer vector of generated tokens
Arguments
- model
WhisperModel
- encoder_output
Encoder hidden states
- initial_tokens
Initial token tensor
- tokenizer
Tokenizer
- max_length
Maximum output length
- device
Device