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