f9b5b2395e4fb5a95e93af0c06d4918ee8410b7a,tools/extract_embeddings.py,,main,#,28

Before Change

    embeddings = onmt.Models.build_embeddings(
                model_opt, tgt_dict.stoi[onmt.IO.PAD_WORD],
                len(tgt_dict), for_encoder=False)
    decoder = onmt.Models.Decoder(model_opt, embeddings)
    encoder_embeddings = encoder.word_lut.weight.data.tolist()
    decoder_embeddings = decoder.word_lut.weight.data.tolist()

    print("Writing source embeddings")

After Change

    embeddings = onmt.Models.build_embeddings(
                model_opt, tgt_dict.stoi[onmt.IO.PAD_WORD],
                len(tgt_dict), for_encoder=False)
    decoder = onmt.Models.Decoder(model_opt.decoder_type, model_opt.rnn_type,
                                  model_opt.dec_layers, model_opt.rnn_size,
                                  model_opt.input_feed,
                                  model_opt.global_attention,
                                  model_opt.coverage_attn,
                                  model_opt.context_gate,
                                  model_opt.copy_attn,
                                  model_opt.dropout, embeddings)

    encoder_embeddings = encoder.word_lut.weight.data.tolist()
    decoder_embeddings = decoder.word_lut.weight.data.tolist()

Instances