c8523b2dc735a1b82ca6170e6ca349defe9f77fc,examples/transformer.py,,,#,22

Before Change

    src, tgt = sess.run([src_text, tgt_text])
    print("src:{}".format(src))
    print("tgt:{}".format(tgt))
    encoder_output = encoder(src_text["text_ids"][:, :-1],
            sequence_length=src_text["length"]-1)
    // Decode
    outputs, final_state, sequence_lengths = decoder(
        initial_state=connector(text_database._hparams.batch_size))

After Change

    src_text = text_data_batch["source_text_ids"]
    tgt_text = text_data_batch["target_text_ids"]

    decoder_inputs = tf.concat((tf.ones_like(tgt_text[:, :1]), tgt_text[:, :-1]), -1)

    print("src_text:{}".format(src_text))
    encoder_output = encoder(src_text,

Instances