Protocols

Module #

Bases: Protocol[P, OutT]

Small protocol that can be used to annotate the input/output types of torch.nn.Modules.

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.

Bases: DataModule[BatchType], Protocol

Protocol that matches "datamodules with a 'num_classes' int attribute.