Algorithm
StepOutputDict #
Algorithm #
Bases: Module
, Protocol[BatchType, StepOutputType]
Protocol that adds more type information to the lightning.LightningModule
class.
This adds some type information on top of the LightningModule class, namely:
- BatchType
: The type of batch that is produced by the dataloaders of the datamodule
- StepOutputType
, the output type created by the step methods.
The networks themselves are created separately and passed as a constructor argument. This is meant to make it easier to compare different learning algorithms on the same network architecture.