You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

13 lines
294 B

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: ...