e9aea97df1dc7878827ac193ba75cbea0b3ee351,ludwig/models/modules/sequence_decoders.py,SequenceGeneratorDecoder,__init__,#SequenceGeneratorDecoder#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,32

Before Change

        self.decoder_cell = LSTMCell(state_size)

        if attention_mechanism:
            if attention_mechanism == "bahdanau":
                pass
            elif attention_mechanism == "luong":
                self.attention_mechanism = tfa.seq2seq.LuongAttention(
                    state_size,
                    None,  // todo tf2: confirm on need
                    memory_sequence_length=max_sequence_length  // todo tf2: confirm inputs or output seq length
                )
            else:
                raise ValueError(
                    "Attention specificaiton "{}" is invalid.  Valid values are "
                    ""bahdanau" or "luong".".format(self.attention_mechanism))

            self.decoder_cell = tfa.seq2seq.AttentionWrapper(
                self.decoder_cell,
                self.attention_mechanism
            )

After Change

        self.decoder_embedding = tf.keras.layers.Embedding(
            input_dim=output_vocab_size,
            output_dim=embedding_dims)
        self.dense_layer = tf.keras.layers.Dense(output_vocab_size)
        self.decoder_rnncell = tf.keras.layers.LSTMCell(rnn_units)

        // Sampler
        self.sampler = tfa.seq2seq.sampler.TrainingSampler()

Instances