d5a75ba09ae0595ca284fd806475eb8d3f771387,word_language_model/generate.py,,,#,18

Before Change

if args.cuda:
    input = input.cuda()

temperature = max(args.temperature, 1e-3)
with open(args.outf, "w") as outf:
    for i in range(args.nwords):

        output, hidden = model(Variable(input, volatile=True), hidden)
        gen = torch.multinomial(output[0].data.div(temperature).exp().cpu(), 1)[0][0] // FIXME: multinomial is only for CPU
        input.fill_(gen)
        word = corpus.dic.idx2word[gen]
        outf.write(word)

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

After Change

    else:
        torch.cuda.manual_seed(args.seed)

if args.temperature < 1e-3:
    parser.error("--temperature has to be greater or equal 1e-3")

with open(args.checkpoint, "rb") as f:
    model = torch.load(f)

if args.cuda:

Instances