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

Before Change

        }
    }
    // Construct the database
    src_db, tgt_db = database.PairedTextDataBase(data_hparams)
    // Get data minibatch, which is a dictionary:
    // {
    //   "text": text_tensor,     // text string minibatch,

After Change

    src_text = text_data_batch["source_text"]
    tgt_text = text_data_batch["target_text"]

    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))

Instances