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