Registry

Module: generative_models.models.registry

Source: generative_models/models/registry.py

Overview

Registry for mapping configuration types to model implementations.

Classes

ModelRegistry

class ModelRegistry

Functions

init

def __init__()

create_model

def create_model()

create_model

def create_model()

decorator

def decorator()

get_factory

def get_factory()

get_model_factory

def get_model_factory()

list_models

def list_models()

list_registered_models

def list_registered_models()

register

def register()

register_model

def register_model()

register_model_legacy

def register_model_legacy()

Module Statistics

• Classes: 1
• Functions: 11
• Imports: 2