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

    print("src_text:{}".format(src_text))
    encoder_output = encoder(src_text,
            sequence_length=text_data_batch["source_length"])
    // Decode
    print("encoder_output:{}".format(encoder_output.shape))
    logits, preds = decoder(decoder_inputs, encoder_output)

Instances