cerebras.modelzoo.common.registry.Registry#
- class cerebras.modelzoo.common.registry.Registry[source]#
Bases:
object
Methods
get_config_class
get_config_class_from_model
get_data_config
get_loss_class
get_model_class
get_path
get_run_path
get_submodel_config_class
list_dataset
list_datasetprocessor
list_loss
list_lr_scheduler
list_models
This method is added to register config classes
This method is added to register config classes
This method is added to register dataset
This method is added to register datasetprocessor
This method is added to register loss
This method is added to register lr_schedular
This method is added to register models
This method is register paths useful for the user
Look for run path for the model
This method is added to register config classes for submodels
This method is added to unregister
Attributes
mapping
submodel_mapping
- classmethod register_model(model_name, datasetprocessor=[], dataset=[])[source]#
This method is added to register models
- classmethod register_datasetprocessor(name)[source]#
This method is added to register datasetprocessor