register_model_builder

hybrid_learning.experimentation.model_registry.model_registry.register_model_builder(model_key, model_getter)

Register a model builder to later be callable via get_model. The builder function must accept all entries of an experiment config dictionary and return a torch model.

Parameters

• model_key (str) –

• model_getter (Callable[[Dict[str, Any]], Module]) –