Sign in
android
/
platform
/
external
/
pytorch
/
c5de535bc0b785abbacfebddf660af4cd3b2a6a1
/
.
/
torch
/
optim
/
sparse_adam.pyi
blob: bc8e4775644284075fb11b11e02eb06f0fdd5bd9 [
file
] [
log
] [
blame
]
from
typing
import
Tuple
from
.
optimizer
import
_params_t
,
Optimizer
class
SparseAdam
(
Optimizer
):
def
__init__
(
self
,
params
:
_params_t
,
lr
:
float
=...,
betas
:
Tuple
[
float
,
float
]=...,
eps
:
float
=...)
->
None
:
...