diff --git a/tensorflow_addons/seq2seq/beam_search_decoder.py b/tensorflow_addons/seq2seq/beam_search_decoder.py index 982370bee8..bbea93463a 100644 --- a/tensorflow_addons/seq2seq/beam_search_decoder.py +++ b/tensorflow_addons/seq2seq/beam_search_decoder.py @@ -762,7 +762,7 @@ def output_dtype(self): parent_ids=tf.int32) def call(self, - embeddning, + embedding, start_tokens, end_token, initial_state, @@ -780,7 +780,7 @@ def call(self, parallel_iterations=self.parallel_iterations, swap_memory=self.swap_memory, training=training, - decoder_init_input=embeddning, + decoder_init_input=embedding, decoder_init_kwargs=init_kwargs)