sgd.pyi 219 B

1234
  1. from .optimizer import _params_t, Optimizer
  2. class SGD(Optimizer):
  3. def __init__(self, params: _params_t, lr: float, momentum: float=..., dampening: float=..., weight_decay:float=..., nesterov:bool=...) -> None: ...