File size: 209 Bytes
345ee20 |
1 2 3 4 5 |
from .base import StepLRScheduler, CosineLRScheduler, WarmupCosineLRScheduler, WarmupPolyLRScheduler
def lr_scheduler_entry(config):
return globals()[config['type']+'LRScheduler'](**config['kwargs'])
|