Skip to content

Protocols

Module #

Bases: Protocol[P, OutT]

Small protocol used to help annotate the input/outputs of torch.nn.Modules.

DataModule #

Bases: Protocol[BatchType]

Protocol that shows the minimal attributes / methods of the LightningDataModule class.

This is used to type hint the batches that are yielded by the DataLoaders.