blob: 6f8db42d3e1733c47cea20fe19e76f66cb98aaf0 [file] [log] [blame]
from .optimizer import _params_t, Optimizer
class SGD(Optimizer):
def __init__(
self,
params: _params_t,
lr: float,
momentum: float = ...,
dampening: float = ...,
weight_decay: float = ...,
nesterov: bool = ...,
) -> None: ...