open3d.ml.torch.dataloaders.ConcatBatcher#

class open3d.ml.torch.dataloaders.ConcatBatcher(device, model='KPConv')#

ConcatBatcher for KPConv.

__init__(device, model='KPConv')#

Initialize.

Parameters:

device – torch device ‘gpu’ or ‘cpu’

Returns:

The corresponding class.

Return type:

class

collate_fn(batches)#

Collate function called by original PyTorch dataloader.

Parameters:

batches – a batch of data

Returns:

the batched result

Return type:

class