register

A register for models

Functions

Summary

get_model

get_model gets a model from the register.

known_model

known_model returns True if the model_type is in the register.

known_model_names

register_model

register_model registers a model.