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
-