Sign in
android
/
platform
/
external
/
pytorch
/
bef7d650c4
/
.
/
torch
/
optim
/
rmsprop.pyi
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
:
...