feat: initial HSAP platform
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>
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
import torch
|
||||
|
||||
|
||||
def build_optimizer(cfg, net):
|
||||
params = []
|
||||
cfg_cp = cfg.optimizer.copy()
|
||||
cfg_type = cfg_cp.pop('type')
|
||||
|
||||
if cfg_type not in dir(torch.optim):
|
||||
raise ValueError("{} is not defined.".format(cfg_type))
|
||||
|
||||
_optim = getattr(torch.optim, cfg_type)
|
||||
return _optim(net.parameters(), **cfg_cp)
|
||||
Reference in New Issue
Block a user