Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
16 lines
358 B
Python
16 lines
358 B
Python
import torch
|
|
import math
|
|
|
|
|
|
def build_scheduler(cfg, optimizer):
|
|
|
|
cfg_cp = cfg.scheduler.copy()
|
|
cfg_type = cfg_cp.pop('type')
|
|
|
|
if cfg_type not in dir(torch.optim.lr_scheduler):
|
|
raise ValueError("{} is not defined.".format(cfg_type))
|
|
|
|
_scheduler = getattr(torch.optim.lr_scheduler, cfg_type)
|
|
|
|
return _scheduler(optimizer, **cfg_cp)
|