sc2bench.models.registry

Functions

get_compressai_model(compression_model_name)	Gets a model in compressai.
get_compression_model(...)	Gets a compression model.
load_classification_model(model_config, ...)	Loads an image classification model.
register_compressai_model(cls_or_func)	Registers a compression model class or a function to build a compression model in compressai.
register_compression_model_class(cls)	Registers a compression model class.
register_compression_model_func(func)	Registers a function to build a compression model.