pytorch/torch/optim/sparse_adam.pyi
2023-10-13 16:33:07 +00:00

13 lines
267 B
Python

from typing import Tuple
from .optimizer import Optimizer, ParamsT
class SparseAdam(Optimizer):
def __init__(
self,
params: ParamsT,
lr: float = ...,
betas: Tuple[float, float] = ...,
eps: float = ...,
) -> None: ...