Skip to content

Model Factory

Source

Model creation factory functions.

model_factory(model_configuration)

Deep learning model factory.

Supported models are defined in careamics.config.SupportedArchitecture.

Parameters:

  • model_configuration (Union[UNetModel, VAEModel]) –

    Model configuration.

Returns:

  • Module –

    Model class.

Raises: