a7cd7a15a9baf741f2b4fcd32f3d8055827889b5,examples/word_language_model/generate.py,,,#,10

Before Change

with open(args.outf, "w") as outf:
    for i in range(args.words):
        output, hidden = learner.module_(input, hidden)
        word_weights = output.squeeze().data.div(args.temperature).exp().cpu()
        word_idx = torch.multinomial(word_weights, 1)[0]
        input.data.fill_(word_idx)
        word = corpus.dictionary.idx2word[word_idx]

        outf.write(word + ("\n" if i % 20 == 19 else " "))

After Change

        word_idx, hidden = learner.sample(input=input,
                                          temperature=args.temperature,
                                          hidden=hidden)
        input = inferno.utils.to_var(torch.LongTensor([[word_idx]]),
                                     use_cuda=args.cuda)

        word = corpus.dictionary.idx2word[word_idx]
        outf.write(word + ("\n" if i % 20 == 19 else " "))

Instances