gnn_tracking.models.meta

gnn_tracking.models.meta#

Wrappers and other “meta” models.

Classes#

Sequential

Sequentially apply modules for and take care of hyperparameters.

Module Contents#

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.

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

Forward.

Contents