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