nip.scenario_base.pretrained_models.register_pretrained_model_class

nip.scenario_base.pretrained_models.register_pretrained_model_class#

nip.scenario_base.pretrained_models.register_pretrained_model_class(pretrained_model_cls: type[P]) type[P][source]#

Register a pretrained model class, so it can be built by name.

Parameters:

pretrained_model_cls (type[PretrainedModel]) – The class to register

Returns:

pretrained_model_cls (type[PretrainedModel]) – The class that was registered, unchanged