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