pytorch/torch/optim/sgd.pyi
2023-10-13 16:33:07 +00:00

12 lines
288 B
Python

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