blob: d37b2f16b8fd2417c56ca325a7408c6b3f35208a [file] [log] [blame]
from .optimizer import _params_t, Optimizer
class RMSprop(Optimizer):
def __init__(
self,
params: _params_t,
lr: float = ...,
alpha: float = ...,
eps: float = ...,
weight_decay: float = ...,
momentum: float = ...,
centered: bool = ...,
) -> None: ...