| 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 | |