Sign in
android
/
platform
/
external
/
pytorch
/
ddb0c26511
/
.
/
torch
/
optim
/
rmsprop.pyi
blob: 168d0bd142a7f05f24f40a3a08b467ba1182529d [
file
] [
log
] [
blame
]
from
.
optimizer
import
Optimizer
,
params_t
class
RMSprop
(
Optimizer
):
def
__init__
(
self
,
params
:
params_t
,
lr
:
float
=
...,
alpha
:
float
=
...,
eps
:
float
=
...,
weight_decay
:
float
=
...,
momentum
:
float
=
...,
centered
:
bool
=
...,
)
->
None
:
...