cuvis_ai.utils.nn_config.SGD
- class cuvis_ai.utils.nn_config.SGD(lr: float = 0.001, alpha: float = 0.0001, power_t: float = 0.5, momentum: float = 0.9, nesterov: bool = True)[source]
Bases:
Optimizer
- __init__(lr: float = 0.001, alpha: float = 0.0001, power_t: float = 0.5, momentum: float = 0.9, nesterov: bool = True) None
Methods
__init__
([lr, alpha, power_t, momentum, ...])Attributes
- property args
- property name
- property pytorch_args
- property sklearn_args