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

Before Change

hidden = model.initHidden(1)

input = torch.LongTensor(1,1).fill_(int(math.floor(torch.rand(1)[0] * ntokens)))
if args.cuda:
    input = input.cuda()

After Change

corpus = data.Corpus(args.data)
ntokens = len(corpus.dictionary)
hidden = model.init_hidden(1)
input = Variable(torch.rand(1, 1).mul(ntokens).long(), volatile=True)
if args.cuda:
    input.data = input.data.cuda()

Instances