uniperceiver/losses/build.py

from uniperceiver.utils.registry import Registry

LOSSES_REGISTRY = Registry("LOSSES")
LOSSES_REGISTRY.__doc__ = """
Registry for losses
"""

def build_losses(cfg):
    losses = []
    for name in cfg.LOSSES.NAMES:
        loss = LOSSES_REGISTRY.get(name)(cfg)
        losses.append(loss)
    return losses