def testMakeBatch(self):
x = [
{"input": torch.randn(4), "target": "a"},
{"input": torch.randn(4), "target": "b"},
]
y = transform.makebatch()(x)
self.assertEqual(y["input"].size(), torch.Size([2, 4]))
self.assertEqual(y["target"], ["a", "b"])
def testMakeBatch(self):
x = [
{
'input': torch.randn(4),
'target': "a"
},
{
'input': torch.randn(4),
'target': "b"
},
]
y = transform.makebatch()(x)
self.assertEqual(y['input'].size(), torch.Size([2, 4]))
self.assertEqual(y['target'], ["a", "b"])
def __init__(self,
dataset,
batchsize,
perm=lambda idx, size: idx,
merge=None,
policy='include-last',
filter=lambda sample: True):
super(BatchDataset, self).__init__()
self.dataset = dataset
self.perm = perm
self.batchsize = batchsize
self.policy = policy
self.filter = filter
self.makebatch = transform.makebatch(merge)
len(self)