core/lr_scheduler/__init__.py

from .base import StepLRScheduler, CosineLRScheduler, WarmupCosineLRScheduler, WarmupPolyLRScheduler

def lr_scheduler_entry(config):
    return globals()[config['type']+'LRScheduler'](**config['kwargs'])