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