f53c38e5ea290616a3be50a99755272a12696070,texar/modules/decoders/rnn_decoders.py,AttentionRNNDecoder,__init__,#AttentionRNNDecoder#Any#Any#Any#Any#Any#Any#Any#Any#,310

Before Change

        attn_kwargs = attn_hparams["kwargs"].todict()

        // Parse the "probability_fn" argument
        prob_fn = attn_kwargs.get("probability_fn", None)
        if prob_fn is None:
            prob_fn = tf.nn.softmax
        if not callable(prob_fn):
            prob_fn = get_class(prob_fn)

After Change

        if "probability_fn" in attn_kwargs:
            prob_fn = attn_kwargs["probability_fn"]
            if prob_fn is not None and not callable(prob_fn):
                prob_fn = get_function(
                    prob_fn,
                    ["tensorflow.nn", "tensorflow.contrib.sparsemax",
                     "tensorflow.contrib.seq2seq"])
            attn_kwargs["probability_fn"] = prob_fn

        attn_kwargs.update({
            "memory_sequence_length": memory_sequence_length,

Instances