gnn_tracking.models.meta

Wrappers and other “meta” models.

Module Contents

Classes

Sequential

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.hparams_mixin.HyperparametersMixin

Sequentially apply modules for and take care of hyperparameters.

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

Forward.