2d47f3f4a1ca342baf92c20a3e1d11c9eb80b3d1,word_language_model/main.py,RNNModel,__init__,#RNNModel#Any#Any#Any#Any#,90

Before Change

        // FIXME: is this better than the standard init? probably
        // FIXME: we need better reset_parameters methods in stdlib
        self.encoder.weight.data.uniform_(-initrange, initrange)
        self.decoder.bias.data.fill_(0)
        self.decoder.weight.data.uniform_(-initrange, initrange)

    def __call__(self, hidden, input):
        emb = self.encoder(input)

After Change

def batchify(data, bsz, bptt):
    nbatch = int(math.floor(data.size(0) / bsz / bptt))
    data = data.narrow(0, 0, nbatch * bptt * bsz)
    data = data.view(bsz, -1).t().contiguous()
    if args.cuda:
        data = data.cuda()
    return data

train = batchify(corpus.train, args.batchsize, args.bptt)
valid = batchify(corpus.valid, 10, 1)
test  = batchify(corpus.test,  10, 1)

Instances