models.generator

Model generation tools

Custom types

List for recursive model generation

ListGen: List[LayerGen | ListGen]

List for storing the state of the model

ListState: List[torch.Tensor | None | ListState]

Classes

BackboneGen	Model base generator
BaseConfig	Base class for model configuration generators
BlockGen	Block generator for the network
Head	Head model holder
HeadGen	Model head generator
ModelGen	Base class for model generators
NeckGen	Network neck generator