vak.models.registry

Registry for models.

Makes it possible to register a model declared outside of vak with a decorator, so that the model can be used at runtime.

Functions

model_family(family_class)	Decorator that adds a lightning.LightningModule class to the registry of model families.
register_model(model)	Function that registers a model in the model registry.