File size: 337 Bytes
32b542e |
1 2 3 4 5 6 7 8 9 10 11 12 |
from uniperceiver.utils.registry import Registry
LR_SCHEDULER_REGISTRY = Registry("LR_SCHEDULER")
LR_SCHEDULER_REGISTRY.__doc__ = """
Registry for lr scheduler
"""
def build_lr_scheduler(cfg, optimizer, data_size):
lr_scheduler = LR_SCHEDULER_REGISTRY.get(cfg.LR_SCHEDULER.NAME)(cfg, optimizer, data_size)
return lr_scheduler
|