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,2 @@
|
||||
from .builder import OPTIMIZERS
|
||||
from .torch_optimizer import torch_optimizer
|
||||
@@ -0,0 +1,3 @@
|
||||
from ..registry import SimpleRegistry
|
||||
|
||||
OPTIMIZERS = SimpleRegistry()
|
||||
@@ -0,0 +1,10 @@
|
||||
from torch import optim
|
||||
|
||||
from .builder import OPTIMIZERS
|
||||
|
||||
|
||||
@OPTIMIZERS.register()
|
||||
def torch_optimizer(torch_optim_class, parameters, *args, **kwargs):
|
||||
# A direct mapping
|
||||
|
||||
return getattr(optim, torch_optim_class)(parameters, *args, **kwargs)
|
||||
Reference in New Issue
Block a user