e75bdc08545f8b3fd312efdbdf1cf94fd62e36b1,onmt/tests/test_models.py,TestModel,get_batch,#TestModel#Any#Any#,38

Before Change

    def get_batch(self, source_l=3, bsize=1):
        // len x batch x nfeat
        test_src = Variable(torch.ones(source_l, bsize, 1)).long()
        test_tgt = Variable(torch.ones(source_l, bsize, 1)).long()
        test_length = torch.ones(bsize).fill_(source_l).long()
        return test_src, test_tgt, test_length

After Change

    def get_batch(self, source_l=3, bsize=1):
        // len x batch x nfeat
        test_src = torch.ones(source_l, bsize, 1).long()
        test_tgt = torch.ones(source_l, bsize, 1).long()
        test_length = torch.ones(bsize).fill_(source_l).long()
        return test_src, test_tgt, test_length

Instances