File size: 296 Bytes
32b542e |
1 2 3 4 5 6 7 8 9 10 11 |
from uniperceiver.utils.registry import Registry
EMBEDDING_REGISTRY = Registry("EMBEDDING")
EMBEDDING_REGISTRY.__doc__ = """
Registry for embedding
"""
def build_embeddings(cfg, name):
embeddings = None if name.lower() == "none" else EMBEDDING_REGISTRY.get(name)(cfg)
return embeddings |