2858c7e09b0c1be5262de68a763dc42aa57fe710,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

        else:
            self.vocab_size = self.num_classes

        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

After Change

        print("setting up attention for", attention_mechanism)
        if attention_mechanism is not None:
            if attention_mechanism == "luong":
                self.attention_mechanism =  LuongAttention(units=state_size)
            elif attention_mechanism == "bahdanau":
                self.attention_mechanism = BahdanauAttention(units=state_size)

            self.decoder_rnncell = AttentionWrapper(self.decoder_rnncell,
                                                self.attention_mechanism,
                                                attention_layer_size=state_size)

Instances