model_registry

Description

Module to handle model registering. This is a trick to flexibly exchange models via Sacred named configs without sacred saving the complete model to the config file. Use this for model configs if the config shall stay readable, must not get too large, and your torch model anyways is not about to change.

Usage:

• register functions returning a model via register_model_builder()

• build a model by key using get_model()

Functions

get_model(model_key, config[, layer_infos, ...])	Return the result of a model builder previously registered via register_model.
register_model_builder(model_key, model_getter)	Register a model builder to later be callable via get_model.
registered_model_keys()	Return a list of valid model builder keys.