gnn_tracking.models.meta

Contents

`gnn_tracking.models.meta`#

Wrappers and other “meta” models.

Module Contents#

Classes#

Sequentially apply modules for and take care of hyperparameters.

class gnn_tracking.models.meta.Sequential(layers: list[torch.nn.Module])#

Bases: torch.nn.Module, pytorch_lightning.core.mixins.HyperparametersMixin

Sequentially apply modules for and take care of hyperparameters.

forward(data: torch_geometric.data.Data) → torch_geometric.data.Data#: Forward.