topobench.nn.wrappers.pointcloud package

Wrappers for pointcloud neural networks with automated exports.

class PointcloudWrapper(backbone, **kwargs)

Bases: AbstractWrapper

Wrapper for Pointcloud models.

This wrapper defines the forward pass of the model. The Pointcloud models return the embeddings of the cells of rank 0.

forward(batch)

Forward pass for the Pointcloud wrapper.

Parameters:

batchtorch_geometric.data.Data

Batch object containing the batched data.

Returns:

dict

Dictionary containing the updated model output.

Submodules

• topobench.nn.wrappers.pointcloud.pointcloud_wrapper module
  • AbstractWrapper
    • AbstractWrapper.__init__()
    • AbstractWrapper.forward()
    • AbstractWrapper.residual_connection()
  • PointcloudWrapper
    • PointcloudWrapper.forward()