lightning

Optional PyTorch Lightning integration: modules, datamodule, callbacks, and metrics.

Modules:

• callbacks –

  Lightning callbacks for batch norm momentum scheduling and metric logging.

• datamodule –

  Lightning datamodule wrapping point cloud datasets with the packed-batch collate.

• metrics –

  Evaluation metrics for 3D detection, instance segmentation, and part segmentation.

• module –

  Lightning modules for classification, segmentation, and detection training.